Research Article
Research on Lightweight Pedestrian Detection Method Based on YOLO
@INPROCEEDINGS{10.1007/978-3-031-50580-5_23, author={Kehua Du and Qinjun Zhao and Rongyao Jing and Lei Zhao and Shijian Hu and Shuaibo Song and Weisong Liu}, title={Research on Lightweight Pedestrian Detection Method Based on YOLO}, proceedings={Multimedia Technology and Enhanced Learning. 5th EAI International Conference, ICMTEL 2023, Leicester, UK, April 28-29, 2023, Proceedings, Part IV}, proceedings_a={ICMTEL PART 4}, year={2024}, month={2}, keywords={Deep learning Pedestrian detection YOLO v5 Shufflenet v2}, doi={10.1007/978-3-031-50580-5_23} }- Kehua Du
Qinjun Zhao
Rongyao Jing
Lei Zhao
Shijian Hu
Shuaibo Song
Weisong Liu
Year: 2024
Research on Lightweight Pedestrian Detection Method Based on YOLO
ICMTEL PART 4
Springer
DOI: 10.1007/978-3-031-50580-5_23
Abstract
Aiming at the problems of large size, high calculation cost and slow detection speed of current pedestrian detection models, this paper proposes a lightweight improved pedestrian detection algorithm based on YOLO v5. Firstly, the Shufflenet v2 network is introduced to replace the backbone network of YOLO v5. Then cascade convolution is designed, and the size of the backbone extraction network convolution core is modified to improve the sensing field of the backbone feature extraction network so that more important context features can be separated. Finally, the unnecessary structure of the backbone network is cut to reduce the scale of network parameters and improve the inference speed. In this paper, the INRIA dataset is used for relevant experiments. Through the experimental analysis of the two algorithms, the size of the model, the number of parameters and the reasoning time of the algorithm in this paper are reduced to 50.1%, 48.6% and 64.7% of YOLO v5s model, respectively. In contrast, the average accuracy of the algorithm is only reduced by 2.1%. This algorithm not only guarantees accuracy, but also greatly improves the reasoning speed.
