Electronics Optics & Control, Volume. 31, Issue 11, 75(2024)

Real-Time Detection of Aerial Refueling Drogue Based on Transformer Feature Pyramid

LI Guilin1, LIU Guihua1, CHEN Tao2, DENG Hao1, and TANG Xue1
Author Affiliations
  • 1School of Information Engineering, Southwest University of Science and Technology, Mianyang 621000, China
  • 2China Ordnance Equipment Group Automation Research Institute CO., LTD, Mianyang 621000, China
  • show less

    The real-time detection of aerial refueling drogue is an important prerequisite for the realization of autonomous aerial refueling. Since the existing object detection algorithms in aerial refueling drogue detection is susceptible to environmental interferences and may result in insufficient accuracy, a real-time detection algorithm for aerial refueling drogue based on the Transformer feature pyramid is proposed. Firstly, a new pooling attention-based Transformer feature pyramid structure TPN is proposed for backbone feature fusion to achieve more efficient feature map enhancement. Then, linear attention is used to reduce complexity of the attention mechanism in TPN, and the lightweight detection model DNet-LinTPN is proposed to reduce the memory consumption by 80%. The experimental results on the self-created air refueling drogue dataset show that the TPN-based model outperforms YOLOv7 in terms of accuracy, speed and model size under the same conditions. The lightweight detection model of DNet-LinTPN achieves an accuracy of 93.8%, which is a 9.4 percentage point improvement over YOLOv7-tiny, with a 67.2% reduction in the amount of parameters and a 45.2% reduction in the amount of operations, and the robustness is obviously improved.

    Tools

    Get Citation

    Copy Citation Text

    LI Guilin, LIU Guihua, CHEN Tao, DENG Hao, TANG Xue. Real-Time Detection of Aerial Refueling Drogue Based on Transformer Feature Pyramid[J]. Electronics Optics & Control, 2024, 31(11): 75

    Download Citation

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

    Category:

    Received: Sep. 26, 2023

    Accepted: Jan. 2, 2025

    Published Online: Jan. 2, 2025

    The Author Email:

    DOI:10.3969/j.issn.1671-637x.2024.11.011

    Topics