ISAT_with_sam/README-en.md

ISAT with segment anything

Image segmentation annotation tool with segment anything.

Quick annotate for image segmentation by segment anything

Project updating，suggestions welcome

Demo Video：youtube

演示视频：bilibili

中文 English

INSTALL

1. Run the source code

(1) Create environment

conda create -n ISAT_with_segment_anything python==3.8
conda activate ISAT_with_segment_anything

(2) Install Segment anything

git clone git@github.com:facebookresearch/segment-anything.git
cd segment-anything
pip install -e .
cd ..

(3) Install ISAT_with_segment_anything

git clone https://github.com/yatengLG/ISAT_with_segment_anything.git
cd ISAT_with_segment_anything
pip install -r requirements.txt

(4) Download Segment anything pretrained checkpoint.

Download the checkpoint，and save in the path: ISAT_with_segment_anything/segment_any

• H-checkpoint:sam_vit_h_4b8939.pth

  H checkpoint has best effect, but need more resources.VRAM needs at least 8G.

• L-checkpoint:sam_vit_l_0b3195.pth

  L checkpoint has normal effect and normal resources.VRAM needs at least 7G.

• B-checkpoint:sam_vit_b_01ec64.pth

  B checkpoint has pool effect, but need less resources.VRAM needs at least 6G.

(5) Run

python main.py

Explain

1. The software needs to run well under the GPU and enought VRAM.
2. If you don't have a GPU or don't have enought VARM, please manually draw polygons for labeling by ISAT.