Open AccessPedestrian detection algorithm based on improved muti-scale feature fusion
Author(s) -
Hui Liu,
Keyang Cheng
Publication year - 2021
Publication title -
journal of physics. conference series
Language(s) - English
Resource type - Journals
SCImago Journal Rank - 0.21
H-Index - 85
eISSN - 1742-6596
pISSN - 1742-6588
DOI - 10.1088/1742-6596/2078/1/012008
Subject(s) - pedestrian detection , computer science , convolution (computer science) , feature (linguistics) , artificial intelligence , scale (ratio) , object detection , algorithm , cluster analysis , upsampling , pattern recognition (psychology) , fusion , process (computing) , pedestrian , image (mathematics) , artificial neural network , engineering , linguistics , philosophy , physics , quantum mechanics , transport engineering , operating system
Aiming at the problem of false detection and missed detection of small targets and occluded targets in the process of pedestrian detection, a pedestrian detection algorithm based on improved multi-scale feature fusion is proposed. First, for the YOLOv4 multi-scale feature fusion module PANet, which does not consider the interaction relationship between scales, PANet is improved to reduce the semantic gap between scales, and the attention mechanism is introduced to learn the importance of different layers to strengthen feature fusion; then, dilated convolution is introduced. Dilated convolution reduces the problem of information loss during the downsampling process; finally, the K-means clustering algorithm is used to redesign the anchor box and modify the loss function to detect a single category. The experimental results show that the improved pedestrian detection algorithm in the INRIA and WiderPerson data sets under different congestion conditions, the AP reaches 96.83% and 59.67%, respectively. Compared with the pedestrian detection results of the YOLOv4 model, the algorithm improves by 2.41% and 1.03%, respectively. The problem of false detection and missed detection of small targets and occlusion has been significantly improved.