├── .gitignore ├── .vscode └── launch.json ├── LICENSE ├── README.md ├── batch_inference_time.sh ├── datasets ├── Dataloader_University.py ├── autoaugment.py ├── make_dataloader.py └── queryDataset.py ├── docs ├── Get_started.md ├── Request.md ├── images │ ├── data.jpg │ ├── framework.jpg │ └── model.png └── training_parameters.md ├── evaluateDistance.py ├── evaluateDistance_DifHeight.py ├── evaluateMA.py ├── evaluateMA_dense.py ├── evaluate_RDS.py ├── evaluate_gpu.py ├── heatmap.py ├── losses ├── ArcfaceLoss.py ├── FocalLoss.py ├── TripletLoss.py ├── __Init__.py ├── cal_loss.py └── loss.py ├── models ├── Backbone │ ├── RKNet.py │ ├── __init__.py │ ├── backbone.py │ └── cvt.py ├── Head │ ├── FSRA.py │ ├── GeM.py │ ├── LPN.py │ ├── NeXtVLAD.py │ ├── NetVLAD.py │ ├── SingleBranch.py │ ├── __init__.py │ ├── head.py │ └── utils.py ├── __init__.py ├── model.py └── taskflow.py ├── optimizers └── make_optimizer.py ├── requirments.txt ├── test.py ├── test_hard.py ├── tool ├── SDM@K_analyze.py ├── SDM@K_compare.py ├── applications │ ├── forwardAllSatelliteHub.py │ ├── inference_global.py │ └── inference_neibor.py ├── dataset_preprocess │ ├── 1-generateSatelliteByUav.py │ ├── 2-generate_new_croped_resized_images_difheights.py │ ├── 3-generate_format_testset.py │ ├── TEST-1-downloadALlAndReCut.py │ ├── TEST-2-generateSatelliteHub.py │ ├── TEST-3-preprocess_difHeightTest.py │ ├── TEST-Regenerate_dense_testset.py │ ├── get_property.py │ ├── google_interface.py │ ├── google_interface_2.py │ ├── split_dataset_long_middle_short.py │ ├── utils.py │ └── validation_testset.py ├── get_inference_time.py ├── get_model_flops_params.py ├── get_property.py ├── mount_dist.sh ├── transforms │ ├── rotatetranformtest.py │ └── transform_visual.py ├── utils.py ├── visual │ ├── Times New Roman.ttf │ ├── demo.py │ ├── demo_custom.py │ ├── demo_custom_visualization.py │ ├── draw_SDMcurve.py │ ├── grad_cam.py │ └── heatmap.py └── visual_demo.py ├── train.py └── train_test_local.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/.gitignore -------------------------------------------------------------------------------- /.vscode/launch.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/.vscode/launch.json -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/README.md -------------------------------------------------------------------------------- /batch_inference_time.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/batch_inference_time.sh -------------------------------------------------------------------------------- /datasets/Dataloader_University.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/datasets/Dataloader_University.py -------------------------------------------------------------------------------- /datasets/autoaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/datasets/autoaugment.py -------------------------------------------------------------------------------- /datasets/make_dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/datasets/make_dataloader.py -------------------------------------------------------------------------------- /datasets/queryDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/datasets/queryDataset.py -------------------------------------------------------------------------------- /docs/Get_started.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /docs/Request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/docs/Request.md -------------------------------------------------------------------------------- /docs/images/data.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/docs/images/data.jpg -------------------------------------------------------------------------------- /docs/images/framework.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/docs/images/framework.jpg -------------------------------------------------------------------------------- /docs/images/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/docs/images/model.png -------------------------------------------------------------------------------- /docs/training_parameters.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/docs/training_parameters.md -------------------------------------------------------------------------------- /evaluateDistance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/evaluateDistance.py -------------------------------------------------------------------------------- /evaluateDistance_DifHeight.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/evaluateDistance_DifHeight.py -------------------------------------------------------------------------------- /evaluateMA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/evaluateMA.py -------------------------------------------------------------------------------- /evaluateMA_dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/evaluateMA_dense.py -------------------------------------------------------------------------------- /evaluate_RDS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/evaluate_RDS.py -------------------------------------------------------------------------------- /evaluate_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/evaluate_gpu.py -------------------------------------------------------------------------------- /heatmap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/heatmap.py -------------------------------------------------------------------------------- /losses/ArcfaceLoss.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /losses/FocalLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/losses/FocalLoss.py -------------------------------------------------------------------------------- /losses/TripletLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/losses/TripletLoss.py -------------------------------------------------------------------------------- /losses/__Init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /losses/cal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/losses/cal_loss.py -------------------------------------------------------------------------------- /losses/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/losses/loss.py -------------------------------------------------------------------------------- /models/Backbone/RKNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/models/Backbone/RKNet.py -------------------------------------------------------------------------------- /models/Backbone/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/Backbone/backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/models/Backbone/backbone.py -------------------------------------------------------------------------------- /models/Backbone/cvt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/models/Backbone/cvt.py -------------------------------------------------------------------------------- /models/Head/FSRA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/models/Head/FSRA.py -------------------------------------------------------------------------------- /models/Head/GeM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/models/Head/GeM.py -------------------------------------------------------------------------------- /models/Head/LPN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/models/Head/LPN.py -------------------------------------------------------------------------------- /models/Head/NeXtVLAD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/models/Head/NeXtVLAD.py -------------------------------------------------------------------------------- /models/Head/NetVLAD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/models/Head/NetVLAD.py -------------------------------------------------------------------------------- /models/Head/SingleBranch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/models/Head/SingleBranch.py -------------------------------------------------------------------------------- /models/Head/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/Head/head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/models/Head/head.py -------------------------------------------------------------------------------- /models/Head/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/models/Head/utils.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/models/model.py -------------------------------------------------------------------------------- /models/taskflow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/models/taskflow.py -------------------------------------------------------------------------------- /optimizers/make_optimizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/optimizers/make_optimizer.py -------------------------------------------------------------------------------- /requirments.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/requirments.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/test.py -------------------------------------------------------------------------------- /test_hard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/test_hard.py -------------------------------------------------------------------------------- /tool/SDM@K_analyze.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/SDM@K_analyze.py -------------------------------------------------------------------------------- /tool/SDM@K_compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/SDM@K_compare.py -------------------------------------------------------------------------------- /tool/applications/forwardAllSatelliteHub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/applications/forwardAllSatelliteHub.py -------------------------------------------------------------------------------- /tool/applications/inference_global.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/applications/inference_global.py -------------------------------------------------------------------------------- /tool/applications/inference_neibor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/applications/inference_neibor.py -------------------------------------------------------------------------------- /tool/dataset_preprocess/1-generateSatelliteByUav.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/dataset_preprocess/1-generateSatelliteByUav.py -------------------------------------------------------------------------------- /tool/dataset_preprocess/2-generate_new_croped_resized_images_difheights.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/dataset_preprocess/2-generate_new_croped_resized_images_difheights.py -------------------------------------------------------------------------------- /tool/dataset_preprocess/3-generate_format_testset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/dataset_preprocess/3-generate_format_testset.py -------------------------------------------------------------------------------- /tool/dataset_preprocess/TEST-1-downloadALlAndReCut.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/dataset_preprocess/TEST-1-downloadALlAndReCut.py -------------------------------------------------------------------------------- /tool/dataset_preprocess/TEST-2-generateSatelliteHub.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/dataset_preprocess/TEST-2-generateSatelliteHub.py -------------------------------------------------------------------------------- /tool/dataset_preprocess/TEST-3-preprocess_difHeightTest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/dataset_preprocess/TEST-3-preprocess_difHeightTest.py -------------------------------------------------------------------------------- /tool/dataset_preprocess/TEST-Regenerate_dense_testset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/dataset_preprocess/TEST-Regenerate_dense_testset.py -------------------------------------------------------------------------------- /tool/dataset_preprocess/get_property.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/dataset_preprocess/get_property.py -------------------------------------------------------------------------------- /tool/dataset_preprocess/google_interface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/dataset_preprocess/google_interface.py -------------------------------------------------------------------------------- /tool/dataset_preprocess/google_interface_2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/dataset_preprocess/google_interface_2.py -------------------------------------------------------------------------------- /tool/dataset_preprocess/split_dataset_long_middle_short.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/dataset_preprocess/split_dataset_long_middle_short.py -------------------------------------------------------------------------------- /tool/dataset_preprocess/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/dataset_preprocess/utils.py -------------------------------------------------------------------------------- /tool/dataset_preprocess/validation_testset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/dataset_preprocess/validation_testset.py -------------------------------------------------------------------------------- /tool/get_inference_time.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/get_inference_time.py -------------------------------------------------------------------------------- /tool/get_model_flops_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/get_model_flops_params.py -------------------------------------------------------------------------------- /tool/get_property.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/get_property.py -------------------------------------------------------------------------------- /tool/mount_dist.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/mount_dist.sh -------------------------------------------------------------------------------- /tool/transforms/rotatetranformtest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/transforms/rotatetranformtest.py -------------------------------------------------------------------------------- /tool/transforms/transform_visual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/transforms/transform_visual.py -------------------------------------------------------------------------------- /tool/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/utils.py -------------------------------------------------------------------------------- /tool/visual/Times New Roman.ttf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/visual/Times New Roman.ttf -------------------------------------------------------------------------------- /tool/visual/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/visual/demo.py -------------------------------------------------------------------------------- /tool/visual/demo_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/visual/demo_custom.py -------------------------------------------------------------------------------- /tool/visual/demo_custom_visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/visual/demo_custom_visualization.py -------------------------------------------------------------------------------- /tool/visual/draw_SDMcurve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/visual/draw_SDMcurve.py -------------------------------------------------------------------------------- /tool/visual/grad_cam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/visual/grad_cam.py -------------------------------------------------------------------------------- /tool/visual/heatmap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/visual/heatmap.py -------------------------------------------------------------------------------- /tool/visual_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/tool/visual_demo.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/train.py -------------------------------------------------------------------------------- /train_test_local.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Dmmm1997/DenseUAV/HEAD/train_test_local.sh --------------------------------------------------------------------------------