Effective Rotate: Learning Rotation-Robust Prototype for Aerial Object Detection
Chaowei Wang,Guangqian Guo,Chang Liu,Dian Shao,Shan Gao
DOI: https://doi.org/10.1109/tgrs.2024.3374880
IF: 8.2
2024-04-03
IEEE Transactions on Geoscience and Remote Sensing
Abstract:Aerial images often depict objects with arbitrary orientations, which pose challenges for conventional object detectors to detect and classify. To address this issue, rotation-equivariant convolutional neural networks (CNNs) have been proposed to extract rotation-equivariant features. However, the orientation encoding in these networks is often unstable and noisy, deteriorating detection performance. In this article, we first analyze the rotation-equivariant network. Then, we propose a rotation-robust prototype generation (RPG) method, which consists of two parts, a stabilization module and an enhancement module. In the stabilization module, we generate rotation-robust prototypes to increase the stability of cyclic shifts. In the enhancement module, we use the obtained prototype to improve the response of the features to object semantics. The RPG method can be used as a plug-and-play module in both one-stage and two-stage detectors. With only 30 lines of code, we achieve an average 1% improvement on four challenging datasets, including DOTA-V1.5, DOTA-v1.0, DIOR-R, and HRSC2016.
imaging science & photographic technology,remote sensing,engineering, electrical & electronic,geochemistry & geophysics