Laser & Optoelectronics Progress, Volume. 60, Issue 12, 1215003(2023)

Attention Mechanism-Based Object Detection Algorithm in Aerial Images

Zongbao Bai1, Junju Zhang1、*, Yuan Gao2, and Youcheng Hu1
Author Affiliations
  • 1School of Electronic and Optical Engineering, Nanjing University of Science & Technology, Nanjing 210094, Jiangsu, China
  • 2School of Electronic and Optical Engineering, Nanjing University of Science and Technology ZiJin College, Nanjing 210023, Jiangsu, China
  • show less

    A target detection algorithm for aerial images based on an improved attention mechanism is suggested to address the issue that the existing object detection network based on horizontal view images has a high false-positive rate and a high miss rate in aerial images. First, a trident channel and spatial attention module that extracts multi-mode and multi-scale characteristic map data of three-branch pooling layers and three-branch dilated convolution layers is added at the output of the Faster R-CNN backbone network so as to compress the data, thereby redistributing the weight of feature channels and spatial pixel regions. Second, a double-head detection mechanism is employed for the classification of the objects and bounding box regression in the aerial image to fully utilize the semantic and spatial location information. The suggested algorithm is further assessed on relevant datasets and contrasted with other object detection algorithms. The results indicate a significant enhancement of the mean average precision of the suggest algorithm, leading to better target detection for unmanned aerial vehicle images in various scenes.

    Tools

    Get Citation

    Copy Citation Text

    Zongbao Bai, Junju Zhang, Yuan Gao, Youcheng Hu. Attention Mechanism-Based Object Detection Algorithm in Aerial Images[J]. Laser & Optoelectronics Progress, 2023, 60(12): 1215003

    Download Citation

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

    Category: Machine Vision

    Received: Mar. 16, 2022

    Accepted: Jun. 13, 2022

    Published Online: Jun. 5, 2023

    The Author Email: Zhang Junju (zj_w1231@163.com)

    DOI:10.3788/LOP221025

    Topics