Computer and Modernization ›› 2024, Vol. 0 ›› Issue (08): 108-113.doi: 10.3969/j.issn.1006-2475.2024.08.017

Previous Articles     Next Articles

Vehicle Detection in UAV Image Based on YOLOv5s

  

  1. (1. School of Automation and Information Engineering, Sichuan University of Science & Engineering, Yibin 644002, China;
    2. Artificial Intelligence Key Laboratory of Sichuan Province, Sichuan University of Science & Engineering, Yibin 644002, China)
  • Online:2024-08-28 Published:2024-08-29

Abstract: The problem of complex backgrounds and large variations in target scales in vehicle images captured by unmanned aerial vehicle (UAV) makes it difficult for existing neural network models to detect small target objects when performing vehicle detection, which can easily lead to false detection and missed detection of small target objects. To solve this problem, an improveed method based on the YOLOv5s neural network is proposed. Firstly, we use the K-means++ algorithm to cluster dataset to obtain better anchor. Secondly, the SPD-Conv small target detection module is combined to reduce the false detection and miss detection rate, so as to improve the accuracy of vehicle detection. Finally, the detection head module is replaced by a decoupled head module to decouple the classification and regression tasks, thus further improve the classification accuracy. The article uses VisDrone-2019-DET dataset for vehicle detection, the mean average precision (mAP) of the improved network in this paper reaches 53.0%, which is 6.3 percentage points higher than the original YOLOv5s model, and can effectively reduce the probability of false detection and missed detection of small objects, enable more accurate vehicle detection.

Key words: YOLOv5s, small object, vehicle detection, K-means++, SPD-Conv, decoupled head model

CLC Number: