An Improved Lightweight Dense Pedestrian Detection Algorithm

Abstract

Due to the limited memory and computing resources in the real application of target detection, the method is challenging to implement on mobile and embedded devices. In order to achieve the balance between detection accuracy and speed in pedestrian-intensive scenes, an improved lightweight dense pedestrian detection algorithm GS-YOLOv5 (GhostNet GSConv- SIoU) is proposed in this paper. In the Backbone section, GhostNet is used to replace the original CSPDarknet53 network structure, reducing the number of parameters and computation. The CBL module is replaced with GSConv in the Head section, and the CSP module is replaced with VoV-GSCSP. The SloU loss function is used to replace the original IoU loss function to improve the prediction box overlap problem in dense scenes. The model parameters are reduced by 40% and the calculation amount is reduced by 64% without losing the average accuracy, and the detection accuracy is improved by 0.5%. The experimental results show that the GS-YOLOv5 can detect pedestrians more effectively under limited hardware conditions to cope with dense pedestrian scenes, and it is suitable for the online real-time detection of pedestrians.