├── .gitignore ├── DataLoader.py ├── LICENSE ├── README.md ├── app.ipynb ├── assets ├── README.md ├── SAM-Med2D_wechat_group.jpeg ├── SAM-Med2D_wechat_group.jpg ├── SAM_Med2D_wechat_group.png ├── cover_SA-Med2D-20M.png ├── dataset.png ├── framwork.png ├── result.png └── visualization.png ├── data_demo ├── image2label_train.json ├── images │ ├── amos_0004_75.png │ ├── amos_0006_90.png │ ├── amos_0507_31.png │ ├── s0114_111.png │ └── s0619_32.png ├── label2image_test.json └── masks │ ├── amos_0004_75_aorta_000.png │ ├── amos_0004_75_inferior_vena_cava_000.png │ ├── amos_0004_75_liver_000.png │ ├── amos_0006_90_aorta_000.png │ ├── amos_0006_90_inferior_vena_cava_000.png │ ├── amos_0006_90_liver_000.png │ ├── amos_0006_90_spleen_000.png │ ├── s0114_111_aorta_000.png │ ├── s0114_111_autochthon_left_000.png │ ├── s0114_111_autochthon_right_000.png │ ├── s0114_111_heart_atrium_left_000.png │ ├── s0114_111_heart_atrium_right_000.png │ ├── s0114_111_heart_myocardium_000.png │ ├── s0114_111_heart_ventricle_left_000.png │ ├── s0114_111_heart_ventricle_right_000.png │ ├── s0114_111_lung_lower_lobe_left_000.png │ ├── s0114_111_lung_lower_lobe_right_000.png │ ├── s0114_111_lung_middle_lobe_right_000.png │ ├── s0114_111_lung_upper_lobe_left_000.png │ ├── s0114_111_rib_left_9_000.png │ ├── s0114_111_rib_right_9_000.png │ ├── s0114_111_vertebrae_T9_000.png │ ├── s0619_32_colon_000.png │ ├── s0619_32_femur_right_000.png │ ├── s0619_32_gluteus_maximus_left_000.png │ ├── s0619_32_gluteus_maximus_right_000.png │ ├── s0619_32_hip_left_000.png │ ├── s0619_32_hip_left_001.png │ ├── s0619_32_hip_right_000.png │ └── s0619_32_hip_right_001.png ├── examples └── SAM-Med2D-onnxruntime │ └── main.py ├── metrics.py ├── predictor_example.ipynb ├── scripts ├── amg.py ├── export_onnx_encoder_model.py └── export_onnx_model.py ├── 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 │ ├── sam_model.py │ └── transformer.py ├── predictor.py ├── predictor_sammed.py └── utils │ ├── __init__.py │ ├── amg.py │ ├── onnx.py │ └── transforms.py ├── test.py ├── train.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/.gitignore -------------------------------------------------------------------------------- /DataLoader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/DataLoader.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/README.md -------------------------------------------------------------------------------- /app.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/app.ipynb -------------------------------------------------------------------------------- /assets/README.md: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /assets/SAM-Med2D_wechat_group.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/assets/SAM-Med2D_wechat_group.jpeg -------------------------------------------------------------------------------- /assets/SAM-Med2D_wechat_group.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/assets/SAM-Med2D_wechat_group.jpg -------------------------------------------------------------------------------- /assets/SAM_Med2D_wechat_group.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/assets/SAM_Med2D_wechat_group.png -------------------------------------------------------------------------------- /assets/cover_SA-Med2D-20M.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/assets/cover_SA-Med2D-20M.png -------------------------------------------------------------------------------- /assets/dataset.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/assets/dataset.png -------------------------------------------------------------------------------- /assets/framwork.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/assets/framwork.png -------------------------------------------------------------------------------- /assets/result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/assets/result.png -------------------------------------------------------------------------------- /assets/visualization.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/assets/visualization.png -------------------------------------------------------------------------------- /data_demo/image2label_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/image2label_train.json -------------------------------------------------------------------------------- /data_demo/images/amos_0004_75.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/images/amos_0004_75.png -------------------------------------------------------------------------------- /data_demo/images/amos_0006_90.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/images/amos_0006_90.png -------------------------------------------------------------------------------- /data_demo/images/amos_0507_31.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/images/amos_0507_31.png -------------------------------------------------------------------------------- /data_demo/images/s0114_111.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/images/s0114_111.png -------------------------------------------------------------------------------- /data_demo/images/s0619_32.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/images/s0619_32.png -------------------------------------------------------------------------------- /data_demo/label2image_test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/label2image_test.json -------------------------------------------------------------------------------- /data_demo/masks/amos_0004_75_aorta_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/amos_0004_75_aorta_000.png -------------------------------------------------------------------------------- /data_demo/masks/amos_0004_75_inferior_vena_cava_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/amos_0004_75_inferior_vena_cava_000.png -------------------------------------------------------------------------------- /data_demo/masks/amos_0004_75_liver_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/amos_0004_75_liver_000.png -------------------------------------------------------------------------------- /data_demo/masks/amos_0006_90_aorta_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/amos_0006_90_aorta_000.png -------------------------------------------------------------------------------- /data_demo/masks/amos_0006_90_inferior_vena_cava_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/amos_0006_90_inferior_vena_cava_000.png -------------------------------------------------------------------------------- /data_demo/masks/amos_0006_90_liver_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/amos_0006_90_liver_000.png -------------------------------------------------------------------------------- /data_demo/masks/amos_0006_90_spleen_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/amos_0006_90_spleen_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0114_111_aorta_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0114_111_aorta_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0114_111_autochthon_left_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0114_111_autochthon_left_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0114_111_autochthon_right_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0114_111_autochthon_right_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0114_111_heart_atrium_left_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0114_111_heart_atrium_left_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0114_111_heart_atrium_right_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0114_111_heart_atrium_right_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0114_111_heart_myocardium_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0114_111_heart_myocardium_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0114_111_heart_ventricle_left_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0114_111_heart_ventricle_left_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0114_111_heart_ventricle_right_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0114_111_heart_ventricle_right_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0114_111_lung_lower_lobe_left_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0114_111_lung_lower_lobe_left_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0114_111_lung_lower_lobe_right_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0114_111_lung_lower_lobe_right_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0114_111_lung_middle_lobe_right_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0114_111_lung_middle_lobe_right_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0114_111_lung_upper_lobe_left_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0114_111_lung_upper_lobe_left_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0114_111_rib_left_9_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0114_111_rib_left_9_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0114_111_rib_right_9_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0114_111_rib_right_9_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0114_111_vertebrae_T9_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0114_111_vertebrae_T9_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0619_32_colon_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0619_32_colon_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0619_32_femur_right_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0619_32_femur_right_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0619_32_gluteus_maximus_left_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0619_32_gluteus_maximus_left_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0619_32_gluteus_maximus_right_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0619_32_gluteus_maximus_right_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0619_32_hip_left_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0619_32_hip_left_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0619_32_hip_left_001.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0619_32_hip_left_001.png -------------------------------------------------------------------------------- /data_demo/masks/s0619_32_hip_right_000.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0619_32_hip_right_000.png -------------------------------------------------------------------------------- /data_demo/masks/s0619_32_hip_right_001.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/data_demo/masks/s0619_32_hip_right_001.png -------------------------------------------------------------------------------- /examples/SAM-Med2D-onnxruntime/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/examples/SAM-Med2D-onnxruntime/main.py -------------------------------------------------------------------------------- /metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/metrics.py -------------------------------------------------------------------------------- /predictor_example.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/predictor_example.ipynb -------------------------------------------------------------------------------- /scripts/amg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/scripts/amg.py -------------------------------------------------------------------------------- /scripts/export_onnx_encoder_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/scripts/export_onnx_encoder_model.py -------------------------------------------------------------------------------- /scripts/export_onnx_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/scripts/export_onnx_model.py -------------------------------------------------------------------------------- /segment_anything/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/segment_anything/__init__.py -------------------------------------------------------------------------------- /segment_anything/automatic_mask_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/segment_anything/automatic_mask_generator.py -------------------------------------------------------------------------------- /segment_anything/build_sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/segment_anything/build_sam.py -------------------------------------------------------------------------------- /segment_anything/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/segment_anything/modeling/__init__.py -------------------------------------------------------------------------------- /segment_anything/modeling/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/segment_anything/modeling/common.py -------------------------------------------------------------------------------- /segment_anything/modeling/image_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/segment_anything/modeling/image_encoder.py -------------------------------------------------------------------------------- /segment_anything/modeling/mask_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/segment_anything/modeling/mask_decoder.py -------------------------------------------------------------------------------- /segment_anything/modeling/prompt_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/segment_anything/modeling/prompt_encoder.py -------------------------------------------------------------------------------- /segment_anything/modeling/sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/segment_anything/modeling/sam.py -------------------------------------------------------------------------------- /segment_anything/modeling/sam_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/segment_anything/modeling/sam_model.py -------------------------------------------------------------------------------- /segment_anything/modeling/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/segment_anything/modeling/transformer.py -------------------------------------------------------------------------------- /segment_anything/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/segment_anything/predictor.py -------------------------------------------------------------------------------- /segment_anything/predictor_sammed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/segment_anything/predictor_sammed.py -------------------------------------------------------------------------------- /segment_anything/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/segment_anything/utils/__init__.py -------------------------------------------------------------------------------- /segment_anything/utils/amg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/segment_anything/utils/amg.py -------------------------------------------------------------------------------- /segment_anything/utils/onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/segment_anything/utils/onnx.py -------------------------------------------------------------------------------- /segment_anything/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/segment_anything/utils/transforms.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OpenGVLab/SAM-Med2D/HEAD/utils.py --------------------------------------------------------------------------------