This is the official implementation of our paper titled "Small Object Detection via Coarse-to-fine Proposal Generation and Imitation Learning", which has been accepted by ICCV 2023.

• CUDA 11.3
• Python 3.8
• PyTorch 1.10.0
• TorchVision 0.11.0
• mmcv-full 1.5.0
• numpy 1.22.4

Our work is based on the large-scale small object detection benchmark SODA, which comprises two sub datasets SODA-D and SODA-A. Download the dataset(s) from corresponding links below.

• SODA-D: OneDrvie; BaiduNetDisk
• SODA-A: OneDrvie; BaiduNetDisk

The data preparation for SODA differs slightly from that of conventional object detection datasets, as it requires the initial step of splitting the original images. Srcipts to obtain sub-images of SODA-D can be found at tools/img_split. For SODA-A, please refer to SODA-mmrotate. More details about SODA please refer to the Dataset Homepage.

This repository is build on MMDetection 2.26.0 which can be installed by running the following scripts. Please ensure that all dependencies have been satisfied before setting up the environment.

git clone https://github.com/shaunyuan22/CFINet
cd CFINet
pip install -v -e .

Moreover, please refer to SODA-mmrotate for MMRotate installation if you want to perform evaluation on the SODA-A dataset.

• Single GPU:

python ./tools/train.py ${CONFIG_FILE} 

• Multiple GPUs:

bash ./tools/dist_train.sh ${CONFIG_FILE} ${GPU_NUM}

• Single GPU:

python ./tools/test.py ${CONFIG_FILE} ${WORK_DIR} --eval bbox

• Multiple GPUs:

bash ./tools/dist_test.sh ${CONFIG_FILE} ${WORK_DIR} ${GPU_NUM} --eval bbox

Please cite our work if you find our work and codes helpful for your research.

@InProceedings{CFINet,
    author    = {Yuan, Xiang and Cheng, Gong and Yan, Kebing and Zeng, Qinghua and Han, Junwei},
    title     = {Small Object Detection via Coarse-to-fine Proposal Generation and Imitation Learning},
    booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)},
    month     = {October},
    year      = {2023},
    pages     = {6317-6327}
}

@ARTICLE{SODA,
  author={Cheng, Gong and Yuan, Xiang and Yao, Xiwen and Yan, Kebing and Zeng, Qinghua and Xie, Xingxing and Han, Junwei},
  journal={IEEE Transactions on Pattern Analysis and Machine Intelligence}, 
  title={Towards Large-Scale Small Object Detection: Survey and Benchmarks}, 
  year={2023},
  volume={45},
  number={11},
  pages={13467-13488}
}

If you have any problems about this repo or SODA benchmark, please be free to contact us at [email protected] 😉