Frontiers of Data and Computing ›› 2023, Vol. 5 ›› Issue (6): 161-172.

CSTR: 32002.14.jfdc.CN10-1649/TP.2023.06.015

doi: 10.11871/jfdc.issn.2096-742X.2023.06.015

Previous Articles     Next Articles

Application of Improved Lightweight YOLOv5 Algorithm in Pedestrian Detection

WANG Ziyuan(),WANG Guozhong*()   

  1. School of Electrical and Electronic Engineering, Shanghai University of Engineering Science, Shanghai 201620, China
  • Received:2022-07-11 Online:2023-12-20 Published:2023-12-25

Abstract:

[Objective] In this paper, we propose an improved YOLOv5 algorithm to address the problems of the high computational complexity of pedestrian detection algorithms, low detection accuracy, and slow detection speed, which can be better applied to pedestrian detection. [Methods] Firstly, the vanilla convolution in the YOLOv5 backbone network is replaced by the depthwise separable convolution, which reduces the number of calculations and parameters while improving detection accuracy. Then, channel attention and spatial attention are incorporated into the feature fusion part of the backbone network, which can force our network to focus on the location and channel information of pedestrians in the image. Finally, the EIOU loss function is used to optimize the proposed model, and the K-means++ clustering algorithm is used to generate priori boxes. [Results] The results show our proposed model can achieve a detection accuracy of 89%, which is 7.6% higher than the original backbone, and the detection speed reaches 106 frames per second when using the INRIA pedestrian detection dataset. [Conclusions] Our proposed method significantly improves the speed and accuracy of pedestrian detection, has also small parameters and is easier to detect and deploy in real-time.

Key words: pedestrian detection, deep learning, YOLOv5, deep separable convolution, attention mechanism