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

Ying Zeng,Yushi Chen,Xue Yang,Qingyun Li,Junchi Yan
DOI: https://doi.org/10.1109/tgrs.2024.3364713
IF: 8.2
2024-02-27
IEEE Transactions on Geoscience and Remote Sensing
Abstract:Existing oriented object detection in aerial images has progressed a lot in recent years and achieved favorable success. However, high-precision oriented object detection in aerial images remains a challenging task. Some recent works have adopted the classification-based method to predict the angle to address boundary problems in angle. However, we have found that these works often neglect the sensitivity of objects with different aspect ratios to angle. At the same time, it is worth exploring a suitable way to improve the emerging transformer-based approaches to adapt them to oriented object detection. In this article, we propose an Aspect Ratio-Sensitive DEtection TRansformer, termed ARS-DETR, for oriented object detection in aerial images. Specifically, a new angle classification method, called aspect ratio-aware circle smooth label (AR-CSL), is proposed to smooth the angle label more reasonably and discard the hyperparameter introduced by previous work [e.g., circular smooth label (CSL)]. Then, a rotated deformable attention (RDA) 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 demonstrate that our method achieves a competitive performance in the high-precision oriented object detection task.
imaging science & photographic technology,remote sensing,engineering, electrical & electronic,geochemistry & geophysics
What problem does this paper attempt to address?