Helmet wearing detection method based on lightweight YOLOv5

Abstract

A lightweight detection algorithm based on YOLOv5 is proposed to solve the problems that the target detection algorithm has many network parameters and the large size of the model is not conducive to the actual deployment. Firstly, the depth separable convolution was introduced to reconstruct the backbone network to reduce the network parameters. Then, the residual element was redesigned by ghost convolution and integrated into the Neck end to further reduce the model volume. Finally, the detection accuracy and recognition effect of the algorithm were improved by modifying the detection layer. When the recognition accuracy of the improved lightweight algorithm is reduced by 3.4%, the model volume is reduced by 75%, and the number of network parameters is reduced by 81%. The results show that the algorithm is able to maintain a high recognition accuracy, and the volume and parameters are greatly reduced. It is suitable for embedded and other system resource limited scenarios, and has a practical application prospect.