Laser & Optoelectronics Progress, Volume. 58, Issue 20, 2010013(2021)

Aerial Image Target Detection Algorithm Based on Improved CenterNet

Yanlei Xu1, Jiran Liang1,2、*, Guojun Dong3, and Zhuang Chen1
Author Affiliations
  • 1School of Microelectronics, Tianjin University, Tianjin 300072, China
  • 2Tianjin Key Laboratory of Imaging and Sensing Microelectronic Technology, Tianjin 300072, China
  • 3Tianjin 712 Communication & Broadcasting Shareholding Co., Ltd., Tianjin 300457, China;
  • show less

    In order to improve the accuracy and speed of aerial image target detection, an improved CenterNet aerial image target detection algorithm based on adaptive threshold is proposed. The center point of the target is used as the key point to replace the anchor box for classification and boundary regression, and an adaptive threshold prediction branch is designed to screen and optimize the preprocessing results. At the same time, the encoding-decoding network structure is designed. Through the deformable cavity convolution structure and the convolutional block attention-connection structure based on the attention mechanism, shallow spatial information, and deep semantic information are effectively extracted and fused. In addition, data enhancement is realized by discarding structured information and building new samples with false and missing detection targets, so as to reduce false and missing detection rates. Experiments are performed on the open data set NWPU VHR-10, the results show that compared with CenterNet based on ResNet-50, mean average precision of proposed algorithm increased by 5.17%, and intersection of union of 0.50 and 0.75 are improved by 3.57% and 3.61%, respectively. The detection speed reaches 45 frame·s -1, achieving good detection accuracy and real-time balance.

    Tools

    Get Citation

    Copy Citation Text

    Yanlei Xu, Jiran Liang, Guojun Dong, Zhuang Chen. Aerial Image Target Detection Algorithm Based on Improved CenterNet[J]. Laser & Optoelectronics Progress, 2021, 58(20): 2010013

    Download Citation

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

    Category: Image Processing

    Received: Dec. 2, 2020

    Accepted: Jan. 6, 2021

    Published Online: Oct. 13, 2021

    The Author Email: Liang Jiran (liang_jiran@tju.edu.cn)

    DOI:10.3788/LOP202158.2010013

    Topics