├── .gitignore ├── .idea ├── .gitignore ├── .name ├── Sample4Geo-main.iml ├── inspectionProfiles │ ├── Project_Default.xml │ └── profiles_settings.xml ├── misc.xml └── modules.xml ├── LPN ├── eval_lpn.py ├── lpn.py ├── result │ └── args.yaml └── settings.yaml ├── README.md ├── evaluate_all.py ├── evaluate_all_cls.py ├── images └── architecture.png ├── model ├── ConvNext.py ├── Head │ ├── __init__.py │ ├── head.py │ └── utils.py ├── SAFA.py ├── __init__.py ├── dataset │ ├── augmentations.py │ ├── denseuav.py │ ├── denseuav_cls.py │ ├── sues.py │ ├── uav_visloc.py │ ├── university.py │ └── visloc.py ├── dino.py ├── dino_utils.py ├── evaluate │ ├── cvusa_and_cvact.py │ ├── eval.py │ ├── eval_cls.py │ ├── university.py │ ├── vigor.py │ └── visloc.py ├── loss.py ├── loss │ ├── FocalLoss.py │ ├── TripletLoss.py │ ├── __init__.py │ ├── cls.py │ └── infonce.py ├── model.py ├── model_cls.py ├── trainer.py ├── trainer_cls.py ├── transforms.py └── utils.py ├── requirements.txt ├── search.py ├── tools ├── compute_MDGM.py ├── compute_MDGM_cls.py ├── get_model_flops_params.py ├── get_model_inference_time.py ├── google_map_downloader.py ├── grad_cam_convnext.py ├── grad_cam_mamba.py ├── grad_cam_resnet50.py ├── grad_cam_vit.py ├── process_datasets.py ├── split_uav.py └── visualization.py ├── train_cls_ddp.py └── train_ddp.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/.gitignore -------------------------------------------------------------------------------- /.idea/.gitignore: -------------------------------------------------------------------------------- 1 | # Default ignored files 2 | /shelf/ 3 | /workspace.xml 4 | -------------------------------------------------------------------------------- /.idea/.name: -------------------------------------------------------------------------------- 1 | eval.py -------------------------------------------------------------------------------- /.idea/Sample4Geo-main.iml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/.idea/Sample4Geo-main.iml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/Project_Default.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/.idea/inspectionProfiles/Project_Default.xml -------------------------------------------------------------------------------- /.idea/inspectionProfiles/profiles_settings.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/.idea/inspectionProfiles/profiles_settings.xml -------------------------------------------------------------------------------- /.idea/misc.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/.idea/misc.xml -------------------------------------------------------------------------------- /.idea/modules.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/.idea/modules.xml -------------------------------------------------------------------------------- /LPN/eval_lpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/LPN/eval_lpn.py -------------------------------------------------------------------------------- /LPN/lpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/LPN/lpn.py -------------------------------------------------------------------------------- /LPN/result/args.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/LPN/result/args.yaml -------------------------------------------------------------------------------- /LPN/settings.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/LPN/settings.yaml -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/README.md -------------------------------------------------------------------------------- /evaluate_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/evaluate_all.py -------------------------------------------------------------------------------- /evaluate_all_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/evaluate_all_cls.py -------------------------------------------------------------------------------- /images/architecture.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/images/architecture.png -------------------------------------------------------------------------------- /model/ConvNext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/ConvNext.py -------------------------------------------------------------------------------- /model/Head/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/Head/head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/Head/head.py -------------------------------------------------------------------------------- /model/Head/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/Head/utils.py -------------------------------------------------------------------------------- /model/SAFA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/SAFA.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/dataset/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/dataset/augmentations.py -------------------------------------------------------------------------------- /model/dataset/denseuav.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/dataset/denseuav.py -------------------------------------------------------------------------------- /model/dataset/denseuav_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/dataset/denseuav_cls.py -------------------------------------------------------------------------------- /model/dataset/sues.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/dataset/sues.py -------------------------------------------------------------------------------- /model/dataset/uav_visloc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/dataset/uav_visloc.py -------------------------------------------------------------------------------- /model/dataset/university.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/dataset/university.py -------------------------------------------------------------------------------- /model/dataset/visloc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/dataset/visloc.py -------------------------------------------------------------------------------- /model/dino.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/dino.py -------------------------------------------------------------------------------- /model/dino_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/dino_utils.py -------------------------------------------------------------------------------- /model/evaluate/cvusa_and_cvact.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/evaluate/cvusa_and_cvact.py -------------------------------------------------------------------------------- /model/evaluate/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/evaluate/eval.py -------------------------------------------------------------------------------- /model/evaluate/eval_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/evaluate/eval_cls.py -------------------------------------------------------------------------------- /model/evaluate/university.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/evaluate/university.py -------------------------------------------------------------------------------- /model/evaluate/vigor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/evaluate/vigor.py -------------------------------------------------------------------------------- /model/evaluate/visloc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/evaluate/visloc.py -------------------------------------------------------------------------------- /model/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/loss.py -------------------------------------------------------------------------------- /model/loss/FocalLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/loss/FocalLoss.py -------------------------------------------------------------------------------- /model/loss/TripletLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/loss/TripletLoss.py -------------------------------------------------------------------------------- /model/loss/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/loss/cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/loss/cls.py -------------------------------------------------------------------------------- /model/loss/infonce.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/loss/infonce.py -------------------------------------------------------------------------------- /model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/model.py -------------------------------------------------------------------------------- /model/model_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/model_cls.py -------------------------------------------------------------------------------- /model/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/trainer.py -------------------------------------------------------------------------------- /model/trainer_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/trainer_cls.py -------------------------------------------------------------------------------- /model/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/transforms.py -------------------------------------------------------------------------------- /model/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/model/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/requirements.txt -------------------------------------------------------------------------------- /search.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/search.py -------------------------------------------------------------------------------- /tools/compute_MDGM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/tools/compute_MDGM.py -------------------------------------------------------------------------------- /tools/compute_MDGM_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/tools/compute_MDGM_cls.py -------------------------------------------------------------------------------- /tools/get_model_flops_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/tools/get_model_flops_params.py -------------------------------------------------------------------------------- /tools/get_model_inference_time.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/tools/get_model_inference_time.py -------------------------------------------------------------------------------- /tools/google_map_downloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/tools/google_map_downloader.py -------------------------------------------------------------------------------- /tools/grad_cam_convnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/tools/grad_cam_convnext.py -------------------------------------------------------------------------------- /tools/grad_cam_mamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/tools/grad_cam_mamba.py -------------------------------------------------------------------------------- /tools/grad_cam_resnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/tools/grad_cam_resnet50.py -------------------------------------------------------------------------------- /tools/grad_cam_vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/tools/grad_cam_vit.py -------------------------------------------------------------------------------- /tools/process_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/tools/process_datasets.py -------------------------------------------------------------------------------- /tools/split_uav.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/tools/split_uav.py -------------------------------------------------------------------------------- /tools/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/tools/visualization.py -------------------------------------------------------------------------------- /train_cls_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/train_cls_ddp.py -------------------------------------------------------------------------------- /train_ddp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cjl-2000/ComplexUAV/HEAD/train_ddp.py --------------------------------------------------------------------------------