Electronics Optics & Control, Volume. 31, Issue 12, 41(2024)

UAV Aerial Image Object Detection Based on Improved YOLOv5s

NING Tao1, FU Shimo2, CHANG Qing1, and WANG Yaoli1
Author Affiliations
  • 1School of Information and Computer, Taiyuan University of Technology, Taiyuan 030000, China
  • 2Taiyuan Water Supply Design and Research Institute Co. Ltd., Taiyuan 030000, China
  • show less

    Aiming at the problem of high miss rate due to the drastic change of target scale, complex background, small and dense targets from the perspective of UAV, an improved YOLOv5s real-time target detection model is proposed. Firstly, a novel hybrid attention mechanism is introduced and embedded into the backbone network to enhance the extraction of crucial target information. Secondly, a new dense residual pyramid pooling is created to improve network information fusion capabilities while reducing computational cost. Then, a C3-BoT module based on multi-head self-attention mechanism is designed to effectively capture the global contextual information of UAV images. Finally, a specialized layer for detecting extremely small targets is added to the YOLOv5s network, specifically tailored to mitigate the issue of miss rate of small objects. Experimental results on the VisDrone2019 dataset show that the improved model achieves an mAP0.5 of 40.6%, an improvement of 8.1 percentage points over the YOLOv5s baseline model, demonstrating superior detection performance in UAV aerial image tasks.

    Tools

    Get Citation

    Copy Citation Text

    NING Tao, FU Shimo, CHANG Qing, WANG Yaoli. UAV Aerial Image Object Detection Based on Improved YOLOv5s[J]. Electronics Optics & Control, 2024, 31(12): 41

    Download Citation

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

    Category:

    Received: Dec. 11, 2023

    Accepted: Dec. 25, 2024

    Published Online: Dec. 25, 2024

    The Author Email:

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

    Topics