GRA: Detecting Oriented Objects through Group-wise Rotating and Attention

  • 2024-03-19 05:40:43
  • Jiangshan Wang, Yifan Pu, Yizeng Han, Jiayi Guo, Yiru Wang, Xiu Li, Gao Huang
  • 0

Abstract

Oriented object detection, an emerging task in recent years, aims to identifyand locate objects across varied orientations. This requires the detector toaccurately capture the orientation information, which varies significantlywithin and across images. Despite the existing substantial efforts,simultaneously ensuring model effectiveness and parameter efficiency remainschallenging in this scenario. In this paper, we propose a lightweight yeteffective Group-wise Rotating and Attention (GRA) module to replace theconvolution operations in backbone networks for oriented object detection. GRAcan adaptively capture fine-grained features of objects with diverseorientations, comprising two key components: Group-wise Rotating and Group-wiseAttention. Group-wise Rotating first divides the convolution kernel intogroups, where each group extracts different object features by rotating at aspecific angle according to the object orientation. Subsequently, Group-wiseAttention is employed to adaptively enhance the object-related regions in thefeature. The collaborative effort of these components enables GRA toeffectively capture the various orientation information while maintainingparameter efficiency. Extensive experimental results demonstrate thesuperiority of our method. For example, GRA achieves a new state-of-the-art(SOTA) on the DOTA-v2.0 benchmark, while saving the parameters by nearly 50%compared to the previous SOTA method. Code will be released.

 

Quick Read (beta)

loading the full paper ...