Cascade-Rotated-RetinaNet

The codes build rotated cascade retinanet for arbitrary-oriented object detection. It supports the following datasets: DOTA, HRSC2016, ICDAR2013, ICDAR2015, UCAS-AOD, NWPU VHR-10, VOC2007, MSRA-TD500.

Getting Started

Installation

cd $ROOT/utils
sh make.sh

cd $ROOT/datasets/DOTA_devkit
sudo apt-get install swig
swig -c++ -python polyiou.i
python setup.py build_ext --inplace

pip install requirements.txt
pip install git+git://github.com/lehduong/torch-warmup-lr.git

Inference

python demo.py

Train

1. prepare dataset and move it into the $ROOT directory.
2. generate imageset files:

cd $ROOT/datasets
python generate_imageset.py

3. start training:

python train.py

Evaluation

prepare labels, take hrsc for example:

cd $ROOT/datasets/evaluate
python hrsc2gt.py

start evaluation:

python eval.py

Detections

Contact me

Homepage | Github | zhihu