├── .gitignore ├── LICENSE ├── README-cn.md ├── README.md ├── assets ├── 000039.png ├── 000039_before_after_finetune.jpg ├── 000039_finetuned.png ├── point_before.gif ├── sam_finetune_cut.gif └── sam_original_cut.gif ├── data_example └── VOCdevkit │ └── VOC2007 │ ├── Annotations │ ├── 000032.xml │ ├── 000033.xml │ ├── 000039.xml │ ├── 000068.xml │ └── 009436.xml │ ├── ImageSets │ └── Segmentation │ │ ├── test.txt │ │ ├── train.txt │ │ ├── trainval.txt │ │ └── val.txt │ ├── JPEGImages │ ├── 000032.jpg │ ├── 000033.jpg │ ├── 000039.jpg │ ├── 000068.jpg │ └── 009436.jpg │ ├── SegmentationClass │ ├── 000032.png │ ├── 000033.png │ ├── 000039.png │ ├── 000068.png │ └── 009436.png │ └── SegmentationObject │ ├── 000032.png │ ├── 000033.png │ ├── 000039.png │ ├── 000068.png │ └── 009436.png ├── finetune_sam_voc.py ├── predict_show.py ├── requirements.txt ├── segment_anything ├── __init__.py ├── automatic_mask_generator.py ├── build_sam.py ├── modeling │ ├── __init__.py │ ├── common.py │ ├── image_encoder.py │ ├── mask_decoder.py │ ├── prompt_encoder.py │ ├── sam.py │ └── transformer.py ├── predictor.py └── utils │ ├── __init__.py │ ├── amg.py │ ├── onnx.py │ └── transforms.py └── utils ├── __init__.py ├── custom_dataset.py ├── general.py └── loss.py /.gitignore: -------------------------------------------------------------------------------- 1 | 2 | .idea 3 | *.pyc 4 | weights/ 5 | runs/ 6 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/LICENSE -------------------------------------------------------------------------------- /README-cn.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/README-cn.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/README.md -------------------------------------------------------------------------------- /assets/000039.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/assets/000039.png -------------------------------------------------------------------------------- /assets/000039_before_after_finetune.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/assets/000039_before_after_finetune.jpg -------------------------------------------------------------------------------- /assets/000039_finetuned.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/assets/000039_finetuned.png -------------------------------------------------------------------------------- /assets/point_before.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/assets/point_before.gif -------------------------------------------------------------------------------- /assets/sam_finetune_cut.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/assets/sam_finetune_cut.gif -------------------------------------------------------------------------------- /assets/sam_original_cut.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/assets/sam_original_cut.gif -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/Annotations/000032.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/Annotations/000032.xml -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/Annotations/000033.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/Annotations/000033.xml -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/Annotations/000039.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/Annotations/000039.xml -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/Annotations/000068.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/Annotations/000068.xml -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/Annotations/009436.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/Annotations/009436.xml -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/ImageSets/Segmentation/test.txt: -------------------------------------------------------------------------------- 1 | 000068 2 | 009436 -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/ImageSets/Segmentation/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/ImageSets/Segmentation/train.txt -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/ImageSets/Segmentation/trainval.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/ImageSets/Segmentation/trainval.txt -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/ImageSets/Segmentation/val.txt: -------------------------------------------------------------------------------- 1 | 000039 -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/JPEGImages/000032.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/JPEGImages/000032.jpg -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/JPEGImages/000033.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/JPEGImages/000033.jpg -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/JPEGImages/000039.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/JPEGImages/000039.jpg -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/JPEGImages/000068.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/JPEGImages/000068.jpg -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/JPEGImages/009436.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/JPEGImages/009436.jpg -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/SegmentationClass/000032.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/SegmentationClass/000032.png -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/SegmentationClass/000033.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/SegmentationClass/000033.png -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/SegmentationClass/000039.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/SegmentationClass/000039.png -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/SegmentationClass/000068.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/SegmentationClass/000068.png -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/SegmentationClass/009436.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/SegmentationClass/009436.png -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/SegmentationObject/000032.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/SegmentationObject/000032.png -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/SegmentationObject/000033.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/SegmentationObject/000033.png -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/SegmentationObject/000039.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/SegmentationObject/000039.png -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/SegmentationObject/000068.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/SegmentationObject/000068.png -------------------------------------------------------------------------------- /data_example/VOCdevkit/VOC2007/SegmentationObject/009436.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/data_example/VOCdevkit/VOC2007/SegmentationObject/009436.png -------------------------------------------------------------------------------- /finetune_sam_voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/finetune_sam_voc.py -------------------------------------------------------------------------------- /predict_show.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/predict_show.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/requirements.txt -------------------------------------------------------------------------------- /segment_anything/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/segment_anything/__init__.py -------------------------------------------------------------------------------- /segment_anything/automatic_mask_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/segment_anything/automatic_mask_generator.py -------------------------------------------------------------------------------- /segment_anything/build_sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/segment_anything/build_sam.py -------------------------------------------------------------------------------- /segment_anything/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/segment_anything/modeling/__init__.py -------------------------------------------------------------------------------- /segment_anything/modeling/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/segment_anything/modeling/common.py -------------------------------------------------------------------------------- /segment_anything/modeling/image_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/segment_anything/modeling/image_encoder.py -------------------------------------------------------------------------------- /segment_anything/modeling/mask_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/segment_anything/modeling/mask_decoder.py -------------------------------------------------------------------------------- /segment_anything/modeling/prompt_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/segment_anything/modeling/prompt_encoder.py -------------------------------------------------------------------------------- /segment_anything/modeling/sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/segment_anything/modeling/sam.py -------------------------------------------------------------------------------- /segment_anything/modeling/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/segment_anything/modeling/transformer.py -------------------------------------------------------------------------------- /segment_anything/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/segment_anything/predictor.py -------------------------------------------------------------------------------- /segment_anything/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/segment_anything/utils/__init__.py -------------------------------------------------------------------------------- /segment_anything/utils/amg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/segment_anything/utils/amg.py -------------------------------------------------------------------------------- /segment_anything/utils/onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/segment_anything/utils/onnx.py -------------------------------------------------------------------------------- /segment_anything/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/segment_anything/utils/transforms.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/custom_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/utils/custom_dataset.py -------------------------------------------------------------------------------- /utils/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/utils/general.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/xzyun2011/finetune_segment_anything_tutorial/HEAD/utils/loss.py --------------------------------------------------------------------------------