Home

Awesome

UFPMP-Det: Toward Accurate and Efficient Object Detection on Drone Imagery

The repo is the official implementation of UFPMP-Det.

The code of UFP module is at mmdet/core/ufp

The code of MP-Det is at mmdet/models/dense_heads/mp_head.py

The config of our project is at configs/UFPMP-Det

Install

  1. This repo is implemented based on mmdetection. Please install it according to get_start.md.
  2. pip install nltk
    pip install albumentations
    

Quickstart

We provide the Dataset(COCO Format) as follows:

We provide the checkpoint as follows:

Training

This repo is only supposed single GPU.

Prepare

Build by yourself: We provide two data set conversion tools.

# conver VisDrone to COCO
python UFPMP-Det-Tools/build_dataset/VisDrone2COCO.py
# conver UAVDT to COCO
python UFPMP-Det-Tools/build_dataset/UAVDT2COCO.py
# build UFP dataset(VisDrone)
CUDA_VISIBLE_DEVICES=2 python UFPMP-Det-Tools/build_dataset/UFP_VisDrone2COCO.py \
    ./configs/UFPMP-Det/coarse_det.py \
    ./work_dirs/coarse_det/epoch_12.pth \
    xxxxxx/dataset/COCO/images/UAVtrain \
    xxxxxx/dataset/COCO/annotations/instances_UAVtrain_v1.json \
    xxxxxx/dataset/COCO/images/instance_UFP_UAVtrain/ \
    xxxxxx/dataset/COCO/annotations/instance_UFP_UAVtrain.json \
    --txt_path path_to_VisDrone_annotation_dir

Download:

In Quick Start

Train Coarse Detector

CUDA_VISIBLE_DEVICES=0 python tools/train.py ./configs/UFPMP-Det/coarse_det.py

Train MP-Det

CUDA_VISIBLE_DEVICES=0 python tools/train.py ./config/UFPMP-Det/mp_det_res50.py

Test

CUDA_VISIBLE_DEVICES=2 python UFPMP-Det-Tools/eval_script/ufpmp_det_eval.py \
    ./configs/UFPMP-Det/coarse_det.py \
    ./work_dirs/coarse_det/epoch_12.pth \
    ./configs/UFPMP-Det/mp_det_res50.py  \
    ./work_dirs/mp_det_res50/epoch_12.pth \
    XXXXX/dataset/COCO/annotations/instances_UAVval_v1.json \
    XXXXX/dataset/COCO/images/UAVval

Citation

If you find our paper or this project helps your research, please kindly consider citing our paper in your publication.

@inproceedings{ufpmpdet,
  title={UFPMP-Det: Toward Accurate and Efficient Object Detection on Drone Imagery},
  author={Huang, Yecheng and Chen, Jiaxin and Huang, Di},
  booktitle={AAAI Conference on Artificial Intelligence},
  year={2022}
}