An Improved YOLOv7 Lightweight Detection Algorithm for Obscured Pedestrians
Author:
Li Chang1, Wang Yiding2, Liu Xiaoming1
Affiliation:
1. College of Electrical and Control Engineering, North China University of Technology, Beijing 100144, China 2. College of Information Science and Technology, North China University of Technology, Beijing 100144, China
Abstract
The detection algorithm commonly misses obscured pedestrians in traffic scenes with a high pedestrian density because mutual occlusion among pedestrians reduces the prediction box score of the concealed pedestrians. The paper uses the YOLOv7 algorithm as the baseline and makes the following three improvements by investigating the variables influencing the detection method’s performance: First, the backbone network of the YOLOv7 algorithm is replaced with the lightweight feature extraction network Mobilenetv3 since the pedestrian detection algorithm frequently needs to be deployed in driverless mobile, which requires a fast operating speed of the algorithm; second, a high-resolution feature pyramid structure is suggested for the issue of missed detection of hidden pedestrians, which upscales the feature maps generated from the feature pyramid to increase the resolution of the output feature maps and introduces shallow feature maps to strengthen the distinctions between adjacent sub-features to enhance the network’s ability to extract features for the visible area of hidden pedestrians and small-sized pedestrians in order to produce deeper features with greater differentiation for pedestrians; and the third is to suggest a detection head based on an attention mechanism that is employed to lower the confidence level of target neighboring sub-features, lower the quantity of redundant detection boxes, and lower the following NMS computation. The mAP of the suggested approach in this work achieves 89.75%, which is 9.5 percentage points better than the YOLOv7 detection algorithm, according to experiments on the CrowdHuman pedestrian-intensive dataset. The algorithm proposed in this paper can considerably increase the detection performance of the detection algorithm, particularly for obscured pedestrians and small-sized pedestrians in the dataset, according to the experimental effect plots.
Subject
Electrical and Electronic Engineering,Biochemistry,Instrumentation,Atomic and Molecular Physics, and Optics,Analytical Chemistry
Reference34 articles.
1. Traffic light detection based on optimized YOLOv3 algorithm;Sun;Acta Opt. Sin.,2020 2. Wu, B., and Nevatia, R. (2005, January 17–21). Detection of multiple, partially occluded humans in a single image by Bayesian combination of edgelet part detectors. Proceedings of the Tenth IEEE International Conference on Computer Vision (ICCV’05), Beijing, China. 3. Sabzmeydani, P., and Mori, G. (2007, January 17–22). Detecting pedestrians by learning shapelet features. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, Minneapolis, MN, USA. 4. Ross, G., Jeff, D., and Trevor, D. (2014, January 23–28). Rich feature hierarchies for accurate object detection and semantic segmentation. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Columbus, OH, USA. 5. Girshick, R. (2015, January 7–13). Fast R-CNN. Proceedings of the IEEE International Conference on Computer Vision (ICCV), Santiago, Chile.
Cited by
3 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献
|
|