Electronics Optics & Control, Volume. 32, Issue 3, 56(2025)

A Lightweight Target Detection Algorithm for Aerial Images

HE Qitian, LI Weixiang, CHENG Ming, SUN Yuan, and CHEN Chuang
Author Affiliations
  • College of Electrical Engineering and Control Science,Nanjing Tech University,Nanjing 211000,China
  • show less

    In order to solve the problems of complex background,small target objects and difficult model deployment in UAV aerial images,a lightweight target detection algorithm for aerial images is proposed. The lightweight FasterNet module is introduced in the backbone network of YOLOv5m to replace C3 module,and the model parameters is compressed to improve reasoning speed of the model. In the feature fusion network,the improved CBAM_L mechanism is used to focus on capturing small target information in aerial images while improving the target recognition accuracy of the model. The detection head in the detection network is replaced by a decoupled head,which solves the conflict between classification and regression when outputting variables in aerial images; and the loss function in the network is replaced by EIoU,which effectively improves the model regression accuracy. The verification results on the public dataset VisDrone show that the average accuracy mAP@0.5 of the improved model is increased by 0.014,the parameter quantity and the computation cost is respectively reduced to 34.3% and 32.4% of the original model,and the detection speed reaches 77 frames per second. The results show that the proposed algorithm exhibits good performance in both detection accuracy and speed.

    Tools

    Get Citation

    Copy Citation Text

    HE Qitian, LI Weixiang, CHENG Ming, SUN Yuan, CHEN Chuang. A Lightweight Target Detection Algorithm for Aerial Images[J]. Electronics Optics & Control, 2025, 32(3): 56

    Download Citation

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

    Category:

    Received: Feb. 28, 2024

    Accepted: Mar. 21, 2025

    Published Online: Mar. 21, 2025

    The Author Email:

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

    Topics