Laser & Optoelectronics Progress, Volume. 59, Issue 24, 2415005(2022)

Rotation Target Detection Algorithm for Remote Sensing Image Using Attention Mechanism

Yu Zhang, Jie Ma*, Jinwen Cui, Yuehua Zhao, and Hong Liu
Author Affiliations
  • School of Electronics and Information Engineering, Hebei University of Technology, Tianjin, 300401, China
  • show less

    In this paper, we propose a rotating target detection algorithm using the improved YOLOv5m to solve the problems of low accuracy and poor target direction in the rotating target detection of optical remote sensing images. First, we integrate the attention mechanism module into the network to improve the ability of the model to extract important features. Second, we consider the contribution of the feature fusion at each node in the feature fusion module, with the addition of a skip connection with the same feature scale. Finally, the angles are discretized through densely coded labels due to the angles and boundaries in rotation detection. The experimental results show that this algorithm achieves a detection accuracy of 82.75% for a subset of DOTA data, indicating an improvement of 11.73 percentage points compared with the original YOLOv5m network when the model computation is reduced slightly. Furthermore, we achieved a detection accuracy of 88.89% in the HRSC2016 ship dataset. That is, the algorithm can effectively improve the accuracy of the rotation detection of optical remote sensing images.

    Tools

    Get Citation

    Copy Citation Text

    Yu Zhang, Jie Ma, Jinwen Cui, Yuehua Zhao, Hong Liu. Rotation Target Detection Algorithm for Remote Sensing Image Using Attention Mechanism[J]. Laser & Optoelectronics Progress, 2022, 59(24): 2415005

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Machine Vision

    Received: Sep. 15, 2021

    Accepted: Oct. 25, 2021

    Published Online: Jan. 11, 2023

    The Author Email: Ma Jie (jma@hebut.edu.cn)

    DOI:10.3788/LOP202259.2415005

    Topics