Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

ARS-DETR: Aspect Ratio-Sensitive Detection Transformer for Aerial Oriented Object Detection

About

Existing oriented object detection methods commonly use metric AP$_{50}$ to measure the performance of the model. We argue that AP$_{50}$ is inherently unsuitable for oriented object detection due to its large tolerance in angle deviation. Therefore, we advocate using high-precision metric, e.g. AP$_{75}$, to measure the performance of models. In this paper, we propose an Aspect Ratio Sensitive Oriented Object Detector with Transformer, termed ARS-DETR, which exhibits a competitive performance in high-precision oriented object detection. Specifically, a new angle classification method, calling Aspect Ratio aware Circle Smooth Label (AR-CSL), is proposed to smooth the angle label in a more reasonable way and discard the hyperparameter that introduced by previous work (e.g. CSL). Then, a rotated deformable attention module is designed to rotate the sampling points with the corresponding angles and eliminate the misalignment between region features and sampling points. Moreover, a dynamic weight coefficient according to the aspect ratio is adopted to calculate the angle loss. Comprehensive experiments on several challenging datasets show that our method achieves competitive performance on the high-precision oriented object detection task.

Ying Zeng, Yushi Chen, Xue Yang, Qingyun Li, Junchi Yan• 2023

Related benchmarks

TaskDatasetResultRank
Oriented Object DetectionDOTA v1.0 (test)
SV80.19
378
Object DetectionDOTA 1.0 (test)
Plane AP86.61
256
Oriented Object DetectionDIOR-R
mAP5066.12
44
Oriented Object DetectionDIOR-R (test)
APL65.82
28
Oriented Object DetectionDOTA 1.0 (train test)
mAP 5075.5
23
Object DetectionDIOR-R
mAP66.12
21
Oriented Object DetectionRSAR (test)
mAP61.14
9
Showing 7 of 7 rows

Other info

Follow for update