├── .gitignore ├── LICENSE ├── README.md ├── assets └── overview.jpg ├── configs ├── base_config.py ├── config_hrsid.py ├── config_nwpu.py └── config_whu.py ├── data ├── HRSID │ └── Annotations │ │ ├── inshore │ │ ├── inshore_test.json │ │ └── inshore_train.json │ │ └── offshore │ │ ├── offshore_test.json │ │ └── offshore_train.json ├── NWPU │ └── Annotations │ │ ├── NWPU_instances_train.json │ │ └── NWPU_instances_val.json └── WHU │ └── annotations │ ├── WHU_building_test.json │ ├── WHU_building_train.json │ └── WHU_building_val.json ├── datasets ├── HRSID.py ├── NWPU.py ├── WHU.py ├── __init__.py ├── augmentation.py └── tools.py ├── inference.py ├── pretrain └── Where_To_Save_Pretrained_SAM_Checkpoints ├── requirements.txt ├── scripts ├── train_hrsid_pointsam.sh ├── train_hrsid_selftrain.sh ├── train_hrsid_supervise.sh ├── train_nwpu_pointsam.sh ├── train_nwpu_selftrain.sh ├── train_nwpu_supervise.sh ├── train_whu_pointsam.sh ├── train_whu_selftrain.sh └── train_whu_supervise.sh ├── segment_anything ├── __init__.py ├── automatic_mask_generator.py ├── build_sam.py ├── modeling │ ├── __init__.py │ ├── common.py │ ├── image_encoder.py │ ├── image_encoder_adapter.py │ ├── mask_decoder.py │ ├── prompt_encoder.py │ ├── sam.py │ └── transformer.py ├── predictor.py └── utils │ ├── __init__.py │ ├── amg.py │ ├── onnx.py │ └── transforms.py ├── train_pointsam.py ├── train_selftrain.py ├── train_supervise.py └── utils ├── eval_utils.py ├── finch.py ├── losses.py ├── model.py ├── sam_lora.py ├── sample_utils.py ├── tools.py ├── utils.py └── visualize.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/README.md -------------------------------------------------------------------------------- /assets/overview.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/assets/overview.jpg -------------------------------------------------------------------------------- /configs/base_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/configs/base_config.py -------------------------------------------------------------------------------- /configs/config_hrsid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/configs/config_hrsid.py -------------------------------------------------------------------------------- /configs/config_nwpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/configs/config_nwpu.py -------------------------------------------------------------------------------- /configs/config_whu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/configs/config_whu.py -------------------------------------------------------------------------------- /data/HRSID/Annotations/inshore/inshore_test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/data/HRSID/Annotations/inshore/inshore_test.json -------------------------------------------------------------------------------- /data/HRSID/Annotations/inshore/inshore_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/data/HRSID/Annotations/inshore/inshore_train.json -------------------------------------------------------------------------------- /data/HRSID/Annotations/offshore/offshore_test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/data/HRSID/Annotations/offshore/offshore_test.json -------------------------------------------------------------------------------- /data/HRSID/Annotations/offshore/offshore_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/data/HRSID/Annotations/offshore/offshore_train.json -------------------------------------------------------------------------------- /data/NWPU/Annotations/NWPU_instances_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/data/NWPU/Annotations/NWPU_instances_train.json -------------------------------------------------------------------------------- /data/NWPU/Annotations/NWPU_instances_val.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/data/NWPU/Annotations/NWPU_instances_val.json -------------------------------------------------------------------------------- /data/WHU/annotations/WHU_building_test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/data/WHU/annotations/WHU_building_test.json -------------------------------------------------------------------------------- /data/WHU/annotations/WHU_building_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/data/WHU/annotations/WHU_building_train.json -------------------------------------------------------------------------------- /data/WHU/annotations/WHU_building_val.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/data/WHU/annotations/WHU_building_val.json -------------------------------------------------------------------------------- /datasets/HRSID.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/datasets/HRSID.py -------------------------------------------------------------------------------- /datasets/NWPU.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/datasets/NWPU.py -------------------------------------------------------------------------------- /datasets/WHU.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/datasets/WHU.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/datasets/augmentation.py -------------------------------------------------------------------------------- /datasets/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/datasets/tools.py -------------------------------------------------------------------------------- /inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/inference.py -------------------------------------------------------------------------------- /pretrain/Where_To_Save_Pretrained_SAM_Checkpoints: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/train_hrsid_pointsam.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/scripts/train_hrsid_pointsam.sh -------------------------------------------------------------------------------- /scripts/train_hrsid_selftrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/scripts/train_hrsid_selftrain.sh -------------------------------------------------------------------------------- /scripts/train_hrsid_supervise.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/scripts/train_hrsid_supervise.sh -------------------------------------------------------------------------------- /scripts/train_nwpu_pointsam.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/scripts/train_nwpu_pointsam.sh -------------------------------------------------------------------------------- /scripts/train_nwpu_selftrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/scripts/train_nwpu_selftrain.sh -------------------------------------------------------------------------------- /scripts/train_nwpu_supervise.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/scripts/train_nwpu_supervise.sh -------------------------------------------------------------------------------- /scripts/train_whu_pointsam.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/scripts/train_whu_pointsam.sh -------------------------------------------------------------------------------- /scripts/train_whu_selftrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/scripts/train_whu_selftrain.sh -------------------------------------------------------------------------------- /scripts/train_whu_supervise.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/scripts/train_whu_supervise.sh -------------------------------------------------------------------------------- /segment_anything/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/segment_anything/__init__.py -------------------------------------------------------------------------------- /segment_anything/automatic_mask_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/segment_anything/automatic_mask_generator.py -------------------------------------------------------------------------------- /segment_anything/build_sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/segment_anything/build_sam.py -------------------------------------------------------------------------------- /segment_anything/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/segment_anything/modeling/__init__.py -------------------------------------------------------------------------------- /segment_anything/modeling/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/segment_anything/modeling/common.py -------------------------------------------------------------------------------- /segment_anything/modeling/image_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/segment_anything/modeling/image_encoder.py -------------------------------------------------------------------------------- /segment_anything/modeling/image_encoder_adapter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/segment_anything/modeling/image_encoder_adapter.py -------------------------------------------------------------------------------- /segment_anything/modeling/mask_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/segment_anything/modeling/mask_decoder.py -------------------------------------------------------------------------------- /segment_anything/modeling/prompt_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/segment_anything/modeling/prompt_encoder.py -------------------------------------------------------------------------------- /segment_anything/modeling/sam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/segment_anything/modeling/sam.py -------------------------------------------------------------------------------- /segment_anything/modeling/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/segment_anything/modeling/transformer.py -------------------------------------------------------------------------------- /segment_anything/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/segment_anything/predictor.py -------------------------------------------------------------------------------- /segment_anything/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/segment_anything/utils/__init__.py -------------------------------------------------------------------------------- /segment_anything/utils/amg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/segment_anything/utils/amg.py -------------------------------------------------------------------------------- /segment_anything/utils/onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/segment_anything/utils/onnx.py -------------------------------------------------------------------------------- /segment_anything/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/segment_anything/utils/transforms.py -------------------------------------------------------------------------------- /train_pointsam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/train_pointsam.py -------------------------------------------------------------------------------- /train_selftrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/train_selftrain.py -------------------------------------------------------------------------------- /train_supervise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/train_supervise.py -------------------------------------------------------------------------------- /utils/eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/utils/eval_utils.py -------------------------------------------------------------------------------- /utils/finch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/utils/finch.py -------------------------------------------------------------------------------- /utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/utils/losses.py -------------------------------------------------------------------------------- /utils/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/utils/model.py -------------------------------------------------------------------------------- /utils/sam_lora.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/utils/sam_lora.py -------------------------------------------------------------------------------- /utils/sample_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/utils/sample_utils.py -------------------------------------------------------------------------------- /utils/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/utils/tools.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/utils/utils.py -------------------------------------------------------------------------------- /utils/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lans1ng/PointSAM/HEAD/utils/visualize.py --------------------------------------------------------------------------------