Journal of System Simulation ›› 2025, Vol. 37 ›› Issue (2): 551-562.doi: 10.16182/j.issn1004731x.joss.23-1564

• Papers • Previous Articles    

Improved Target Detection Algorithm for Aerial Images Based on YOLOv5

Guo Yecai1,2, Sun Jingdong1, Saha Amitave1   

  1. 1.School of Electronic and Information Engineering, Nanjing University of Information Science & Technology, Nanjing 210044, China
    2.School of Electronic Information Engineering, Wuxi University, Wuxi 214105, China
  • Received:2023-12-25 Revised:2024-01-25 Online:2025-02-14 Published:2025-02-10

Abstract:

In order to improve the existing small target detection methods, which suffer from low detection accuracy, high false detection rate and high leakage rate, the FSD-YOLOv5 algorithm is proposed, which has three improvements based on the YOLOv5 algorithm. The Focal EIoU is used instead of the original CIoU to improve the model convergence speed and regression accuracy. To cope with the deficiencies in CNN architecture, we adopt a new CNN building block called SPD-Conv is adopted. To address the problem of the reduced or lost information of small objects in feature maps caused by downsampling in convolutional neural networks, feature reuse is introduced to increase the feature information of small objects in the feature maps. Experimental results show that FSD-YOLOv5 achieves a detection accuracy of 36.3%, an improvement of 2.4% in comparison with original algorithm.

Key words: YOLOv5, Focal EIoU, SPD-Conv, densenet, aerial image detection

CLC Number: