We have released the code for our paper "Towards Generalized UAV Object Detection: A Novel Perspective from Frequency Domain Disentanglement"，which is accepted as IJCV 2024. It is an extension of our CVPR 2023 paper "Generalized UAV Object Detection via Frequency Domain Disentanglement", with our code using Faster-RCNN as the exemplified detection network.
[[Download link for UAVDT] https://pan.baidu.com/s/1oqfCJr3T_l3wFM9wpmHW7g (pwd: shdl) ]
[[Download link for VisDrone2019] https://pan.baidu.com/s/1KbpgWTIoWojUlTZXHl3gaw (pwd: xbkt) ]
Please refer to the requirements.txt file in the directory, where we have listed all the dependencies required for setting up the environment.
[[Download link] https://pan.baidu.com/s/171LJz9gHNhrAZCzd3ZoDvg (pwd: 5jno) ]
The folder structure should be organized as follows:
├── cfgs
├── data
│ ├── UAVDT_Faster
│ │ ├── Annotations (40409 xml in total)
│ │ ├── ImageSets
│ │ │ ├── Main
│ │ │ │ ├── test_dark.txt
│ │ │ │ ├── test_haze.txt
│ │ │ │ ├── train.txt
│ │ ├── JPEGImages (40409 img in total)
│ ├── VisDrone2019_Faster
│ │ ├── Annotations (8251 xml in total)
│ │ ├── ImageSets
│ │ │ ├── Main
│ │ │ │ ├── test_illumination.txt
│ │ │ │ ├── train.txt
│ │ ├── JPEGImages (8251 img in total)
├── imgs
├── lib
├── output (for evaluation output)
├── model_zoo
│ ├── Faster.pth
│ ├── resnet50.pth
...
cd UAV-Frequency
unzip lib.zip -d lib
mkdir output
mkdir model_zoo
mkdir data
cd lib
python setup.py build develop
cd ..
python test_net.py --cudaIf you find our work interesting, please cite the following:
@inproceedings{wang2023generalized,
title={Generalized UAV Object Detection via Frequency Domain Disentanglement},
author={Wang, Kunyu and Fu, Xueyang and Huang, Yukun and Cao, Chengzhi and Shi, Gege and Zha, Zheng-Jun},
booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
pages={1064--1073},
year={2023}
}
@article{wang2024towards,
title={Towards Generalized UAV Object Detection: A Novel Perspective from Frequency Domain Disentanglement},
author={Wang, Kunyu and Fu, Xueyang and Ge, Chengjie and Cao, Chengzhi and Zha, Zheng-Jun},
journal={International Journal of Computer Vision},
volume={132},
number={11},
pages={5410--5438},
year={2024},
publisher={Springer}
}