• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science ›› 2022, Vol. 44 ›› Issue (08): 1449-1456.

• Graphics and Images • Previous Articles     Next Articles

A complex pedestrian detection model based on improved YOLOv4 algorithm

LI Lan,LIU Jie,ZHANG Jie   

  1. (School of Information and Control Engineering,Qingdao University of Technology,Qingdao 266000,China)
  • Received:2020-12-30 Revised:2021-02-25 Accepted:2022-08-25 Online:2022-08-25 Published:2022-08-25

Abstract: When pedestrian detection is carried out in image and video sequences, there are some problems, such as diverse pedestrian posture and scale as well as pedestrian occlusion, which leads to inaccurate detection of some pedestrians by YOLOv4 algorithm and false detection and missed detection. In order to solve this problem, a complex pedestrian detection model based on improved YOLOv4 algorithm is proposed. Firstly, the ground truth size of pedestrian dataset is analyzed by the improved k-means clustering algorithm, and the size of anchor box is determined according to the clustering results. Secondly, PANet is used for multi-scale feature fusion to make it more sensitive to multi-attitude and multi-scale pedestrian targets and improve the detection effect. Finally, for pedestrian occlusion problem, the repulsion loss function is proposed to make the predicted box as close to the correct target as possible. The experimental results show that the new de-tection model has better detection effect than YOLOv4 and other pedestrian detection model.

Key words: pedestrian detection, YOLOv4 algorithm, k-means clustering algorithm, multi-scale feature fusion, repulsion loss