Journal of Optoelectronics · Laser, Volume. 35, Issue 6, 641(2024)

Lightweight YOLOv5 detection algorithm for low-altitude micro UAV

WEI Feng1, ZHOU Jianping1, TAN Xiang1,2,3、*, LIN Jing2, TIAN Li2, and WANG Hu1
Author Affiliations
  • 1School of Intelligent Manufacturing Modern Industry, Xinjiang University, Urumchi, Xinjiang Uygur Autonomous Region 830000, China
  • 2Institute of Geographic Sciences and Natural Resources Research, Chinese Academy of Sciences, Beijing 100101, China
  • 3The Research Center for UAV Application and Regulation, Chinese Academy of Sciences, Beijing 100101, China
  • show less

    Aiming at the problem that low-altitude micro-UAVs pose a threat to public safety, this paper proposes a lightweight target detection model YOLOv5_SS suitable for mobile terminals based on the you only look once v5 (YOLOv5) network. In this model, the lightweight network ShuffleNetv2 replaces the original backbone network of YOLOv5, introduces squeeze-and-excitation networks (SENet) attention mechanism, and uses soft non-maximum suppression (Soft-NMS) algorithm to improve the detection effect of dense overlapping targets. The experimental results show that the mean average precision@0.5 (mAP50) of the model for the detection of low-altitude micro-UAV on the dataset is 92.75%, the accuracy is 90.49%, and the number of parameters is 0.237 4 M. The number of floating-point operations is 0.9GFLOPS (giga floating-point operations).

    Tools

    Get Citation

    Copy Citation Text

    WEI Feng, ZHOU Jianping, TAN Xiang, LIN Jing, TIAN Li, WANG Hu. Lightweight YOLOv5 detection algorithm for low-altitude micro UAV[J]. Journal of Optoelectronics · Laser, 2024, 35(6): 641

    Download Citation

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

    Category:

    Received: Oct. 28, 2022

    Accepted: Dec. 13, 2024

    Published Online: Dec. 13, 2024

    The Author Email: TAN Xiang (tanxiang@igsnrr.ac.cn)

    DOI:10.16136/j.joel.2024.06.0741

    Topics