├── README.md ├── configs └── settings.py ├── data ├── LoadDataSeg.py ├── coco_train.py ├── coco_val.py ├── remote_train.py ├── remote_val.py ├── transforms │ ├── 22 │ ├── functional.py │ └── transforms.py ├── voc_train.py └── voc_val.py ├── data_list ├── all_train.txt ├── all_val.txt ├── train │ ├── split0_train.txt │ ├── split1_train.txt │ ├── split2_train.txt │ └── split3_train.txt └── val │ ├── split0_val.txt │ ├── split1_val.txt │ ├── split2_val.txt │ └── split3_val.txt ├── img ├── remote_sensing1.png ├── remote_sensing1_1.png ├── remote_sensing_dataset.png └── remote_sensing_result.png ├── models ├── ASPP.py ├── PMMs.py ├── PMMs_single.py └── backbone │ ├── resnet.py │ ├── resnet_dialated.py │ └── vgg.py ├── networks ├── FPMMs.py ├── FPMMs_one.py ├── FPMMs_vgg.py ├── FRPMMs.py ├── FRPMMs_vgg.py ├── PFENet.py ├── R_pmm.py ├── __init__.py ├── canet.py ├── panet.py ├── resnet.py └── vgg.py ├── scripts ├── 001 ├── train_group0.sh ├── train_group1.sh ├── train_group2.sh └── train_group3.sh ├── test.py ├── test_5shot.py ├── test_all_frame.py ├── test_frame.py ├── train.py └── utils ├── NoteEvaluation.py ├── NoteLoss.py ├── Restore.py ├── Visualize.py └── my_optim.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/README.md -------------------------------------------------------------------------------- /configs/settings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/configs/settings.py -------------------------------------------------------------------------------- /data/LoadDataSeg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/data/LoadDataSeg.py -------------------------------------------------------------------------------- /data/coco_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/data/coco_train.py -------------------------------------------------------------------------------- /data/coco_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/data/coco_val.py -------------------------------------------------------------------------------- /data/remote_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/data/remote_train.py -------------------------------------------------------------------------------- /data/remote_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/data/remote_val.py -------------------------------------------------------------------------------- /data/transforms/22: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /data/transforms/functional.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/data/transforms/functional.py -------------------------------------------------------------------------------- /data/transforms/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/data/transforms/transforms.py -------------------------------------------------------------------------------- /data/voc_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/data/voc_train.py -------------------------------------------------------------------------------- /data/voc_val.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/data/voc_val.py -------------------------------------------------------------------------------- /data_list/all_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/data_list/all_train.txt -------------------------------------------------------------------------------- /data_list/all_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/data_list/all_val.txt -------------------------------------------------------------------------------- /data_list/train/split0_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/data_list/train/split0_train.txt -------------------------------------------------------------------------------- /data_list/train/split1_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/data_list/train/split1_train.txt -------------------------------------------------------------------------------- /data_list/train/split2_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/data_list/train/split2_train.txt -------------------------------------------------------------------------------- /data_list/train/split3_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/data_list/train/split3_train.txt -------------------------------------------------------------------------------- /data_list/val/split0_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/data_list/val/split0_val.txt -------------------------------------------------------------------------------- /data_list/val/split1_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/data_list/val/split1_val.txt -------------------------------------------------------------------------------- /data_list/val/split2_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/data_list/val/split2_val.txt -------------------------------------------------------------------------------- /data_list/val/split3_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/data_list/val/split3_val.txt -------------------------------------------------------------------------------- /img/remote_sensing1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/img/remote_sensing1.png -------------------------------------------------------------------------------- /img/remote_sensing1_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/img/remote_sensing1_1.png -------------------------------------------------------------------------------- /img/remote_sensing_dataset.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/img/remote_sensing_dataset.png -------------------------------------------------------------------------------- /img/remote_sensing_result.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/img/remote_sensing_result.png -------------------------------------------------------------------------------- /models/ASPP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/models/ASPP.py -------------------------------------------------------------------------------- /models/PMMs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/models/PMMs.py -------------------------------------------------------------------------------- /models/PMMs_single.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/models/PMMs_single.py -------------------------------------------------------------------------------- /models/backbone/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/models/backbone/resnet.py -------------------------------------------------------------------------------- /models/backbone/resnet_dialated.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/models/backbone/resnet_dialated.py -------------------------------------------------------------------------------- /models/backbone/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/models/backbone/vgg.py -------------------------------------------------------------------------------- /networks/FPMMs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/networks/FPMMs.py -------------------------------------------------------------------------------- /networks/FPMMs_one.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/networks/FPMMs_one.py -------------------------------------------------------------------------------- /networks/FPMMs_vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/networks/FPMMs_vgg.py -------------------------------------------------------------------------------- /networks/FRPMMs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/networks/FRPMMs.py -------------------------------------------------------------------------------- /networks/FRPMMs_vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/networks/FRPMMs_vgg.py -------------------------------------------------------------------------------- /networks/PFENet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/networks/PFENet.py -------------------------------------------------------------------------------- /networks/R_pmm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/networks/R_pmm.py -------------------------------------------------------------------------------- /networks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/networks/__init__.py -------------------------------------------------------------------------------- /networks/canet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/networks/canet.py -------------------------------------------------------------------------------- /networks/panet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/networks/panet.py -------------------------------------------------------------------------------- /networks/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/networks/resnet.py -------------------------------------------------------------------------------- /networks/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/networks/vgg.py -------------------------------------------------------------------------------- /scripts/001: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /scripts/train_group0.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/scripts/train_group0.sh -------------------------------------------------------------------------------- /scripts/train_group1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/scripts/train_group1.sh -------------------------------------------------------------------------------- /scripts/train_group2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/scripts/train_group2.sh -------------------------------------------------------------------------------- /scripts/train_group3.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/scripts/train_group3.sh -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/test.py -------------------------------------------------------------------------------- /test_5shot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/test_5shot.py -------------------------------------------------------------------------------- /test_all_frame.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/test_all_frame.py -------------------------------------------------------------------------------- /test_frame.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/test_frame.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/train.py -------------------------------------------------------------------------------- /utils/NoteEvaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/utils/NoteEvaluation.py -------------------------------------------------------------------------------- /utils/NoteLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/utils/NoteLoss.py -------------------------------------------------------------------------------- /utils/Restore.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/utils/Restore.py -------------------------------------------------------------------------------- /utils/Visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/utils/Visualize.py -------------------------------------------------------------------------------- /utils/my_optim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/caoql98/SDM/HEAD/utils/my_optim.py --------------------------------------------------------------------------------