Home

Awesome

DOTA-DOAI

Abstract

This repo is the codebase for our team to participate in DOTA related competitions, including rotation and horizontal detection. We mainly use FPN-based two-stage detector, and it is completed by Xue Yang and Jirui Yang.

We also recommend two rotated/oriented object detection benchmarks, which are led by Xue Yang and Yue Zhou.

Performance

DOTA1.0 (Task1)

ModelBackboneTraining dataVal datamAPModel LinkTrickslr schdData AugmentationGPUImage/GPUConfigs
FPNResNet152_v1d (600,800,1024)->MSDOTA1.0 trainvalDOTA1.0 test78.99modelALL2xYes2X GeForce RTX 2080 Ti1cfgs_dota1.0_res152_v1.py

DOTA1.0 (Task2)

ModelBackboneTraining dataVal datamAPModel LinkTrickslr schdData AugmentationGPUImage/GPUConfigs
FPN (memory consumption)ResNet152_v1d (600,800,1024)->MSDOTA1.0 trainvalDOTA1.0 test81.23modelALL2xYes2X Quadro RTX 80001cfgs_dota1.0_res152_v1.py

Visualization

1

Performance of published papers on DOTA datasets

DOTA1.0 (Task1)

HBox-supervised ModelBackbonemAPPaper LinkCode LinkRemarkRecommend
H2RBoxResNet5074.40ICLR2023Pytorch, Jittorweak- and self-supervised:white_check_mark:
RBox-supervised ModelBackbonemAPPaper LinkCode LinkRemarkRecommend
FR-O (DOTA)ResNet10152.93CVPR2018MXNetDOTA dataset, baseline:white_check_mark:
IENetResNet10157.14arXiv:1912.00969-anchor free
TOSOResNet10157.52ICASSP2020-geometric transformation
Wang et al.HRNet58.29Comput. Electr. Eng.-anchor free
PIoU LossDLA-3460.5ECCV2020PyTorchIoU loss, anchor free:white_check_mark:
R<sup>2</sup>CNNResNet10160.67arXiv:1706.09579TFscene text, multi-task, different pooled sizes, baseline:white_check_mark:
RRPNResNet10161.01TMM arXiv:1703.01086TFscene text, rotation proposals, baseline:white_check_mark:
Axis LearningResNet10165.98Remote SensingPytorchsingle stage, anchor free:white_check_mark:
Li et al.ResNet5066.01IGARSS2021-refine, feature alignment
MARNetResNet10167.19IJRS-based on scrdet
ICNResNet10168.16ACCV2018-image cascade, multi-scale:white_check_mark:
GSDetResNet10168.28TIP-scale reasoning
RADetResNeXt10169.09Remote Sensing-enhanced FPN, mask rcnn
KARNETResNet5068.87CISNRC 2020-attention denoising, anchor refining
RoI TransformerResNet10169.56CVPR2019MXNet, PyTorch, MMRotateroi transformer:white_check_mark:
CAD-NetResNet10169.90TGRS arXiv:1903.00857-attention
ProbIoUResNet5070.04arXiv:2106.06072TFgaussian bounding boxes, hellinger distance:white_check_mark:
ROTPResNet10170.29arXiv:2202.06565-solar corona heatmap, key points, head of object
A<sup>2</sup>S-DetResNet10170.64Remote Sensing-label assign
AOODResNet10171.18Neural Computing and Applications-attention + R-DFPN
CGP BoxResNet1871.35IJRS-center-guide points
Zhou et al.CSPDarknet5371.5ijgi-enhanced fpn
ACEDLA3471.7TIP-corner point:white_check_mark:
Pei et al.ResNet10171.76IGRASS2021-enhanced FPN
Cascade-FFResNet15271.80ICME2020-refined retinanet + feature fusion
SCPNetHourglass10472.20GRSL-corner points
P-RSDetResNet10172.30Access-anchor free, polar coordinates:white_check_mark:
Zhang et al.ResNet10172.37GSIS-refine-stage
ROPDetResNet101-DCN72.42J REAL-TIME IMAGE PR-point set representation
SCRDetResNet10172.61ICCV2019TF: R<sup>2</sup>CNN++, IoU-Smooth L1: RetinaNet-based, R<sup>3</sup>Det-basedattention, angular boundary problem:white_check_mark:
O<sup>2</sup>-DNetHourglass10472.8ISPRS, arXiv:1912.10694-centernet, anchor free:white_check_mark:
HRPNetHRNet-W4872.83GRSL-polar
SARDResNet10172.95Access-IoU-based weighted loss
GLS-NetResNet10172.96Remote Sensing-attention, saliency pyramid
ProjBBResNet10173.03Accesscode, codebasenew definition of bounding box
DRNHourglass10473.23CVPR2020codecenternet, feature selection module, dynamic refinement head, new dataset (SKU110K-R):white_check_mark:
FADetResNet10173.28ICIP2019-attention
RBA-CenterNetResNet10173.41IJCNN-centernet, refine feature
MFIAR-NetResNet15273.49Sensors-feature attention, enhanced FPN
CFC-NETResNet10173.50TGRSPyTorchcritical feature, label assign, refine:white_check_mark:
Dual-DetResNet1873.62IJRScodekeypoint-based
Li et al.CSP-Hourglass73.70GRSL-CSP-Hourglass Net
R<sup>3</sup>DetResNet10173.79AAAI2021TF, r3det-on-mmdetection, r3det-pytorch, MMRotaterefined single stage, feature alignment:white_check_mark:
SDCDetResNet10173.89PRAI2021-instance segmentation direction correction
Geng et al.ResNet10173.92GRSL-anchor free, angle encoding
Free<sup>3</sup>NetResNet10174.04TMM-anchor free, gliding vertex:white_check_mark:
SegmRDetResNet5074.14Neurocomputing-segmentation-baed, new training and inference
Hou et al.ResNet10174.44TIP-enhanced FPN, feature alignment
Wu et al.ResNet5074.45J Electron Imaging-enhanced FPN, feature alignment
CenterRotResNet15274.75Remote Sensing-anchor free, deformable-fpn, csl
MEADResNet10174.80Applied Intelligence-mechanism anchor free, mask guided, refine feature
TS<sup>4</sup>NetResNet10174.82Neurocomputing-label assign
FEDetResNet5074.89ICCSE 2021-refine feature, angle constraint
SRep-RDetRepVGG-B1g274.89IJRS-refine feature, attention, repvgg
Yuan et al.HRNet3274.97ICCTIT2021-centernet-r
Gliding VertexResNet10175.02TPAMI arXiv:1911.09358PyTorch, MMRotatequadrilateral bbox:white_check_mark:
OSSDetResNeXt10175.08JSTARS-feature enhancement and alignment
LO-DetDarknet5375.24TGRS, arXiv:2209.07709PyTorchlightweight:white_check_mark:
EFNU-Net75.27Preprints-Field-based:white_check_mark:
SARResNet15275.26Access-boundary problem:white_check_mark:
TricubeNetHourglass10475.26WACV2022code2D tricube kernel:white_check_mark:
Mask OBBResNeXt10175.33Remote Sensing-attention, multi-task:white_check_mark:
BBAVectorsResNet10175.36WACV2021PyTorchkeypoint based:white_check_mark:
SAOAResNet10175.41ICIG2021-anchor free, spatial self-attention
Yuan et al.ResNet10175.43TGRS-feature alignment, continuous boundary
Zand et al.DarkNet5375.5TGRS-angle classification
MSFF-75.60ICCECE 2022-multi-scale feature fusion
RODFormerViT-B475.60Sensors-vision transformer
FFAResNet10175.7ISPRS-enhanced FPN, rotation proposals
CBDA-NetDLA-34-DCN75.74TGRS-dual attention
APEResNeXt101(32x4)75.75TGRS arXiv:1906.09447-adaptive period embedding, length independent IoU (LIIoU):white_check_mark:
R<sup>4</sup>DetResNet15275.54Image Vis Comput-feature recursion and refinement
SurroundNetResNet15275.88Remote Sensing-attention, anchor free, quad
RIEHRGANet-W4875.94Remote Sensing-center-based rotated inscribed ellipse
F<sup>3</sup>-NetResNet15276.02Remote Sensing-feature fusion and filtration
CenterMap OBBResNet10176.03TGRS-center-probability-map
DA-NetResNet10176.11GRSL-feature alignment
MDL-pResNet10176.16arXiv:2204.00840-gaussian modeling
CSLResNet15276.17ECCV2020TF, MMRotate, Pytorch: YOLOv5_DOTA_OBB (CSL)angular boundary problem:white_check_mark:
MRDetResNet10176.24TGRS-arbitrary-oriented rpn, multiple subtasks
AFC-NetResNet10176.27Neurocomputing-adaptive feature concatenate
RSDet/RSDet++ResNet15276.30AAAI2021/TCSVTTFquadrilateral bbox, angular boundary problem:white_check_mark:
OWSREnsemble76.36CVPR2019 WorkShop-enhanced FPN
SLAResNet5076.36Remote SensingPyTorchsparse label assignment:white_check_mark:
SE<sup>2</sup>-DetResNet10176.42Remote Sensing-enhanced fpn
OPLDResNet10176.43J-STARSPyTorchboundary problem, point-guided:white_check_mark:
Polar RayResNet10176.50ACM MM2021-polar rays representation:white_check_mark:
SIoUResNet5076.54JSTARS-splicing intersection over union:white_check_mark:
R<sup>3</sup>Det++ResNet15276.56arXiv:2004.13316TFrefined single stage, feature alignment, denoising:white_check_mark:
PolarDetResNet10176.64IJRS arXiv:2010.08720-polar, center-semantic:white_check_mark:
Beyond Bounding-BoxResNet15276.67CVPR2021PyTorch, MMRotatepoint-based, reppoints:white_check_mark:
OANResNeX5076.73arXiv:2212.13136PyTorchobjectness activation network, efficient:white_check_mark:
SCRDet++ResNet10176.81TPAMITFangular boundary problem, denoising:white_check_mark:
DAFNeResNet10176.95arXiv:2109.06148PyTorchsingle stage, anchor free, center-to-corner regression
DAL+S<sup>2</sup>A-NetResNet5076.95AAAI2021PyTorchlabel assign:white_check_mark:
GGHLDarkNet5376.95TIPPyTorchgaussian heatmap labeling:white_check_mark:
EAutoDetDarkNet5377.05arXiv:2203.10747-nas, yolov5:white_check_mark:
Yu et al.Res2Net5077.18JSTARS-boundary-aware vectors, centernet
CoF-NetResNet5077.2TGRS-coarse-to-fine, geometric constraints, spatial-spectral nonocal features
DCLResNet15277.37CVPR2021TFboundary problem:white_check_mark:
CSL+DCLResNet15277.37IJCVTFboundary problem:white_check_mark:
CLT-DetResNet10177.45TGRS-transformer, correlation learning:white_check_mark:
MSFFResNet5077.46JSTARS-rotation invariance features
RIDetResNet5077.62GRSLPyTorch, TFquad., representation ambiguity:white_check_mark:
Oriented RepPointsSwin-Tiny77.63CVPR2022PyTorch, MMRotatepoint-based, reppoints:white_check_mark:
RSPViTAEv2-S77.72arXiv:2204.02825PyTorchremote sensing pretrain:white_check_mark:
RDDResNet10177.75Remote SensingPyTorchrotation-decoupled
CenterOBBDLA-3477.85Remote Sensing-angle classification, centernet:white_check_mark:
FSDetResNet5077.85TGRS-label assign, feature refinemen, anchor free:white_check_mark:
CG-NetResNet10177.89arXiv:2103.11399PyTorchattention
HSPResNet10178.01TGRS-hierarchical semantic propagation
FoRDetVGG1678.13TGRS-refinenet
AProNetResNet10178.16ISPRSPyrotchaxis projection-based angle learning, feature enhancement
MGARDarkNet5378.29arXiv:2209.02884-angle classification:white_check_mark:
TransConvNetSwin Transformer78.41Remote Sensing-enhanced fpn, self attention, transformer
FR-EstResNet101-DCN78.49TGRS-point-based estimator:white_check_mark:
TIOE-Det-78.69ISPRSPyTorchmisaligned tasks, orientation estimation:white_check_mark:
DARDetResNet5078.74GRSLPyTorchvarifocalnet, dcn, piou
TS-ConvDarkNet5378.75arXiv:2209.02200PyTorchdynamic label assignment, task-wise samping:white_check_mark:
DHRecResNet15278.83TPAMIPyTorchdouble horizontal rectangles
FCOSRResNeXt10178.90arXiv:2111.10780PyTorchanchor free
P2PResNet10179.15AAAI2022-poly iou loss:white_check_mark:
SASMResNeXt10179.17AAAI2022MMRotate, PyTorchlabel assign:white_check_mark:
AO2-DETRResNet5079.22arXiv:2205.12785PyTorchdetr, feature refinement:white_check_mark:
GSNetResNet10179.37arXiv:2204.02033-enhanced fpn
S<sup>2</sup>A-NetResNet50/ResNet10179.42/79.15TGRSPyTorch, MMRotaterefined single stage, feature alignment:white_check_mark:
OFA-NetResNet10179.52PRICAI2021-refined single stage, feature alignment
TARDetResNet10179.57CVPRW2022-anchor free, feature alignment
RBoxResNet5079.59CVPR2022-transformer, feature sampling and grouping:white_check_mark:
O<sup>2</sup>DETRResNet5079.66arXiv:2106.03146-deformable detr, transformer:white_check_mark:
ROSDResNet10179.76Access-refined single stage, feature alignment
RAODResNet10179.78Applied Intelligence-enhanced fpn, feature refine
SES-NetResNet5079.80arXiv:2111.03420-sampling equivariance, self-attention:white_check_mark:
SARAResNet50/ResNet10179.91/79.13Remote Sensing-self-adaptive aspect ratio anchor, refine
ARP+R-EIoUYOLOv5x679.93arXiv:2109.10187-area ratio of parallelogram, R-EIoU, yolov5
GF-CSLResNet10179.94TGRSPyTorchcircular smooth label, gaussian focal loss:white_check_mark:
ADT-DetResNet15279.95Remote Sensing-feature pyramid transformer, feature refinement
ReDetReR50-ReFPN80.10CVPR2021PyTorch, MMRotaterotation-equivariant, rotation-invariant roI align:white_check_mark:
G-RepSwin-Tiny80.12arXiv:2205.11796MMRotatepointset-based, gaussian modeling:white_check_mark:
PCIReR50-ReFPN80.15TIP-progressive context-dependent inference:white_check_mark:
GWDResNet15280.23ICML2021TF, MMRotateboundary discontinuity, square-like problem, gaussian wasserstein distance loss:white_check_mark:
DEAReR50-ReFPN80.37TGRSPyTorchdynamic enhancement anchor:white_check_mark:
FDOLReR50-ReFPN80.41TGRS-frequency analysis, self-attention:white_check_mark:
O2MERResNet5080.43arXiv:2112.00504-consistent geometric constraint
DODetResNet5080.62TGRSPyTorchoriented proposal network, localization-guided head
KLDResNet15280.63NeurIPS2021TF, MMRotateKullback-Leibler divergence, high-precision, scale invariance:white_check_mark:
AOPGResNet50/ResNet10180.66/80.19arXiv:2110.01931PyTorchanchor free, feature align
Li et al.ResNet10180.68Remote Sensing-enhance fpn, polar
CGCDetResNet5080.70TNNLSPyTorchgeometric consistent constraint, label assignment:white_check_mark:
AFF-DetResNet5080.73Acm T Multim Comput.-enhanced fpn
PP-YOLOE-R-xCRN-x80.73arXiv:2211.02386Paddlereal-time, yolo:white_check_mark:
Oriented R-CNNResNet50/ResNet10180.87/80.52ICCV2021PyTorch, MMRotateRotation FPN, Gliding Vertex
OSKDetResNet10180.91CVPR2022-keypoint localization:white_check_mark:
KFIoUSwin-Tiny80.93ICLR2023TF, MMRotateGaussian modeling, kalman filter:white_check_mark:
QPDetResNet5081.00TGRSPyTorchquadrant point regression, rotated box refinement:white_check_mark:
Point RCNNReR50-ReFPN/Swin-Tiny80.71/81.32Remote Sensing-point based, cascade rcnn:white_check_mark:
SSEDetResNet5081.08GRSL-circle theorem, short-side excursion:white_check_mark:
RTMDet-81.33arXiv:2212.07784MMRotatereal-time:white_check_mark:

DOTA1.0 (Task2)

ModelBackbonemAPPaper LinkCode LinkRemarkRecommend
FR-H (DOTA)ResNet10160.46CVPR2018MXNetDOTA dataset, baseline:white_check_mark:
Deep Active LearningResNet1864.26arXiv:2003.08793-CenterNet, Deep Active Learning:white_check_mark:
SBLResNet5064.77arXiv:1810.08103-single stage
CenterFPANetResNet1865.29HPCCT & BDAI 2020 arXiv:2009.03063-light-weight
MARNetResNet10171.73IJRS-based on scrdet
FMSSDVGG1672.43TGRS-IoU-based weighted loss, enhanced FPN
ICNResNet10172.45ACCV2018-image cascade, multi-scale:white_check_mark:
IoU-Adaptive R-CNNResNet10172.72Remote Sensing-IoU-based weighted loss, cascade
EFRVGG1673.49Remote SensingPyTorchenhanced FPN
AF-EMSResNet10173.97Remote Sensing-scale-aware feature, anchor free
SCRDetResNet10175.35ICCV2019TFattention, angular boundary problem:white_check_mark:
FADetResNet10175.38ICIP2019-attention
MFIAR-NetResNet15276.07Sensors-feature attention, enhanced FPN
F<sup>3</sup>-NetResNet15276.48Remote Sensing-feature fusion and filtration
Mask OBBResNeXt10176.98Remote Sensing-attention, multi-task:white_check_mark:
CenterMap OBBResNet10177.33TGRS-center-probability-map
ASSDVGG1677.8TGRS-feature aligned
AFC-NetResNet10178.06Neurocomputing-adaptive feature concatenate
CG-NetResNet10178.26arXiv:2103.11399PyTorchattention
OPLDResNet10178.35J-STARSPyTorchboundary problem, point-guided:white_check_mark:
A<sup>2</sup>RMNetResNet10178.45Remote Sensing-attention, enhanced FPN, different pooled sizes
OWSREnsemble78.79CVPR2019 WorkShop-enhanced FPN
NPMMR-DetDarkNet5378.93TGRSPyTorchnon-local, enhanced FPN
Parallel Cascade R-CNNResNeXt10178.96Journal of Physics: Conference Series-cascade rcnn
DM-FPNResNet-Based79.27Remote Sensing-enhanced FPN
DDBNResNet10179.3PR-adjacent feature compensation, dual detection branch
SCRDet++ResNet10179.35TPAMITFdenoising:white_check_mark:
HSPResNet10180.42TGRS-hierarchical semantic propagation
AFF-DetResNet10181.18Acm T Multim Comput.-enhanced fpn

DOTA1.5 (Task1)

RBox-supervised ModelBackbonemAPPaper LinkCode LinkRemarkRecommend
OANResNet5066.95arXiv:2212.13136PyTorchobjectness activation network, efficient:white_check_mark:
DAFNeResNet10171.99arXiv:2109.06148PyTorchr-fcos
RAODResNet10174.70Applied Intelligence-enhanced fpn, feature refine
OWSREnsemble76.60CVPR2019 WorkShop-enhanced FPN
ReDetReR50-ReFPN76.80CVPR2021PyTorchrotation-equivariant, rotation-invariant RoI Align,:white_check_mark:
FDOLReR50-ReFPN77.12TGRS-frequency analysis, self-attention:white_check_mark:
APEResNeXt101(32x4)78.34TGRS arXiv:1906.09447-length independent IoU (LIIoU):white_check_mark:
Point RCNNReR50-ReFPN/Swin-Tiny79.31/80.14Remote Sensing-point based, cascade rcnn:white_check_mark:

DOTA1.5 (Task2)

ModelBackbonemAPPaper LinkCode LinkRemarkRecommend
CDD-NetResNet10161.3GRSL-attention
GCWNetResNet10162.7GRSL-enhance fpn, feature refine
ReDetReR50-ReFPN78.08CVPR2021PyTorchrotation-equivariant, rotation-invariant RoI Align,:white_check_mark:
OWSREnsemble79.50CVPR2019 WorkShop-enhanced FPN

DOTA2.0 (Task1)

RBox-supervised ModelBackbonemAPPaper LinkCode LinkRemarkRecommend
OANResNet5054.68arXiv:2212.13136PyTorchobjectness activation network, efficient:white_check_mark:
Wen et al.-58.22ICCECE 2022-yolov5, csl

Related Articles

ModelPaper LinkCode LinkRemarkRecommend
SSSDETICIP2019 arXiv:1909.00292-vehicle detection, lightweight
AVDNetGRSL arXiv:1907.07477-vehicle detection, small object
ClusDetICCV2019Caffe2object cluster regions:white_check_mark:
DMNetCVPR2020 WorkShop-object cluster regions:white_check_mark:
AdaZoomarXiv:2106.10409-object cluster regions, reinforcement learning:white_check_mark:
OISarXiv:1911.07732related Pytorch codeOriented Instance Segmentation:white_check_mark:
ISOPIGARSS2020-Oriented Instance Segmentation
LR-RCNNarXiv:2005.14264 -vehicle detection-
GRS-DetTGRS-ship detection, rotation fcos-
DRBoxarXiv:1711.09405Caffesar object detection:white_check_mark:
DRBox-v2TGRSTFsar object detection-
RAPiDarXiv:2005.11623PyTorchoverhead fisheye images-
OcSaFPNarXiv:2012.09859-denoising-
CR2A-NetTGRS-ship detection-
-TGRS-knowledge distillation:white_check_mark:
CHPDetarXiv:2101.11189-new ship dataset:white_check_mark:
EDRdetPRL-adaptive dynamic networks, coarse-to-fine:white_check_mark:

Other Rotation Detection Codes and Benchmarks

Base MethodCode Link
RetinaNetRetinaNet_Tensorflow_Rotation
YOLOv3rotate-yolov3-Pytorch, YOLOv3-quadrangle-Pytorch, yolov3-polygon-Pytorch
YOLOv4rotate-yolov4-Pytorch
YOLOv5rotation-yolov5-Pytorch, YOLOv5_DOTA_OBB (CSL)
YOLOXYOLOX-Pytorch
YOLOv7yolov7-obb (KLD)
CenterNetR-CenterNet-Pytorch
JDetJittor
OBBDet_SwinOBBDet_Swin
OrientedRepPointsOrientedRepPoints_DOTA_Swin
Localization DistillationRotated-LD

Dataset

Some remote sensing related object detection dataset statistics are in DATASET.md