Results and models are available in the model zoo.
Supported backbones:
- ResNet
- ResNeXt
- VGG
- HRNet
- RegNet
- Res2Net
Supported oriented detection methods:
- Oriented R-CNN (ICCV'2021)
- Poly IoU Loss
- Faster R-CNN OBB
- Double Head OBB
- RetinaNet OBB
- Gliding Vertex
- RoI Transformer
- FCOS OBB
Supported horizontal detection methods:
- RPN
- Fast R-CNN
- Faster R-CNN
- Mask R-CNN
- Cascade R-CNN
- Cascade Mask R-CNN
- SSD
- RetinaNet
- GHM
- Mask Scoring R-CNN
- Double-Head R-CNN
- Hybrid Task Cascade
- Libra R-CNN
- Guided Anchoring
- FCOS
- RepPoints
- Foveabox
- FreeAnchor
- NAS-FPN
- ATSS
- FSAF
- PAFPN
- Dynamic R-CNN
- PointRend
- CARAFE
- DCNv2
- Group Normalization
- Weight Standardization
- OHEM
- Soft-NMS
- Generalized Attention
- GCNet
- Mixed Precision (FP16) Training
- InstaBoost
- GRoIE
- DetectoRS
- Generalized Focal Loss
Please refer to install.md for installation and dataset preparation.
If you want to train or test a oriented model, please refer to oriented_model_starting.md.
If you are not familiar with MMdetection, please see getting_started.md for the basic usage of MMDetection. There are also tutorials for finetuning models, adding new dataset, designing data pipeline, and adding new modules.
We refered S2ANet and AerialDetection when develping OBBDetection.
This toolbox is modified from MMdetection. If you use this toolbox or benchmark in your research, please cite the following information.
@article{mmdetection,
title = {{MMDetection}: Open MMLab Detection Toolbox and Benchmark},
author = {Chen, Kai and Wang, Jiaqi and Pang, Jiangmiao and Cao, Yuhang and
Xiong, Yu and Li, Xiaoxiao and Sun, Shuyang and Feng, Wansen and
Liu, Ziwei and Xu, Jiarui and Zhang, Zheng and Cheng, Dazhi and
Zhu, Chenchen and Cheng, Tianheng and Zhao, Qijie and Li, Buyu and
Lu, Xin and Zhu, Rui and Wu, Yue and Dai, Jifeng and Wang, Jingdong
and Shi, Jianping and Ouyang, Wanli and Loy, Chen Change and Lin, Dahua},
journal = {arXiv preprint arXiv:1906.07155},
year={2019}
}
This is the official implement of Oriented R-CNN. if it is used in your research, please cite the following information.
@InProceedings{Xie_2021_ICCV,
author = {Xie, Xingxing and Cheng, Gong and Wang, Jiabao and Yao, Xiwen and Han, Junwei},
title = {Oriented R-CNN for Object Detection},
booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)},
month = {October},
year = {2021},
pages = {3520-3529} }