Electronics Optics & Control, Volume. 31, Issue 4, 22(2024)

UAV Image Recognition Based on Improved YOLOv5s

LI Jie1... WANG Feng1, MA Chen2, WU Guorui1, ZHAO Wei2 and KANG Zhiqiang2 |Show fewer author(s)
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • show less

    UAVs can provide target-related image information in such fields as military intelligence and aerial photography detection,providing target information for image processing tasks.An improved UAV image recognition algorithm based on YOLOv5s is proposed to address the issues of complex background,small detection targets and limited extractable features in UAV images.Firstly,the network structure is optimized by using CotNet module to enhance the model's self-learning ability and improve recognition accuracy.Secondly,the Neck network is improved to better utilize the rich information contained in shallow feature maps to locate targets through cross-layer linking and an improvement to feature map resolution.In the detection head section,decoupled detection heads are used to reduce the conflict between localization and classification tasks over feature information utilization in the prediction process.Finally,in order to improve the convergence rate and model accuracy,the aspect ratio of the width to the height of the loss function is optimized based on CIoU and EIoU loss functions.Tests are conducted on the test set of public dataset VisDrone.The proposed algorithm improves mAP50 and mAP50︰95 by 6.1 and 2.9 percentage points respectively in comparison with the original algorithm.The experimental results show that the proposed model can effectively improve the accuracy of UAV image recognition.

    Tools

    Get Citation

    Copy Citation Text

    LI Jie, WANG Feng, MA Chen, WU Guorui, ZHAO Wei, KANG Zhiqiang. UAV Image Recognition Based on Improved YOLOv5s[J]. Electronics Optics & Control, 2024, 31(4): 22

    Download Citation

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

    Category:

    Received: Apr. 27, 2023

    Accepted: --

    Published Online: Jul. 30, 2024

    The Author Email:

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

    Topics