Research on UAV Small Target Detection Algorithm Based on Improved YOLOv5s
DOI:
CSTR:
Author:
Affiliation:

1.School of Mechanical Engineering, Dalian Jiaotong University, Dalian 116028 , China ; 2.School of Automation and ElectricalEngineering, Dalian Jiaotong University, Dalian 116028 , China

Clc Number:

TP391.4

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    Objective】Aiming at the characteristics of various target scales, complex background and dense small targets in aerial images of unmanned aerial vehicles (UAC), a small target detection algorithm LM-YOLO based on YOLOV5 is proposed.【Method】Firstly, the number of small target detection head was increased and K-DBSCAN clustering algorithm was used to optimize the anchor frame, so as to generate an anchor frame more suitable for small target detection and improve the detection accuracy of the algorithm. Then, a more efficient MobileNetV3-CBAM was designed as a feature extraction network to reduce the size of the network model. Finally, the large kernel selective attention mechanism LSK was introduced into the feature fusion network to increase the resolution of the model to similar targets.【Result】The experimental results on the public data set VisDrone2019 show that the average detection accuracy of LM-YOLO for all targets is improved by 7.6% and the model size is reduced by 45% compared with the benchmark model YOLOV5.【Conclusion】Experiments show that the proposed algorithm can reduce the model size while maintaining good detection accuracy.

    Reference
    Related
    Cited by
Get Citation

董华军,王宇栖.基于改进YOLOv5s的无人机小目标检测算法研究[J].华东交通大学学报英文版,2024,41(4):118-126.
Dong Huajun, Wang Yuqi. Research on UAV Small Target Detection Algorithm Based on Improved YOLOv5s[J]. JOURNAL OF EAST CHINA JIAOTONG UNIVERSTTY,2024,41(4):118-126

Copy
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:March 15,2024
  • Revised:
  • Adopted:
  • Online: September 23,2024
  • Published:
Article QR Code