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

Clc Number:

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,a small target detection algorithm LM-YOLO based on YOLOV5 is proposed.【Method】Firstly,increase the small target detection head and K-DBSCAN clustering algorithm is 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 is designed as a feature extraction network to reduce the size of the network model.Finally,the large kernel selective attention mechanism LSK is 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, and is suitable for the target detection task of aerial images.

    Reference
    Related
    Cited by
Get Citation
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:March 15,2024
  • Revised:April 18,2024
  • Adopted:April 25,2024
  • Online:
  • Published:
Article QR Code