SAM-Segmentation-Tool

Simple mask making using SAM, QT UI included.

Installation

• torch cuda envrionment required
• PyQt5
• segment_anything

  pip install git+https://github.com/facebookresearch/segment-anything.git
  pip install opencv-python pycocotools matplotlib onnxruntime onnx
  

• download model weights to your BASE_DIR

Prepare Images

Put your Image that you want to make Masks for to BASE_DIR/Img

Excute

• Excute SST.py

• Select images and click to create Masks

• Select Mask layer that you want to make

  

• Then press Make, .npy file will be stored at BASE_DIR/Mask/(img_name)_mask.npy