YOLOv5s-FCG : An Improved YOLOv5 Method for Inspecting Riders’ Helmet Wearing

Abstract

Manual inspection of riders’ helmets is time-consuming and labor-intensive, low in efficiency and small in coverage. Aiming at this shortcoming, this paper proposed an improved YOLOv5s-FCG(FourLayers, CBAM attention, GhostBottleneck) helmet wearing detection method based on YOLOv5 (You Only Look Once). Based on the smallest volume of YOLOv5s in YOLOv5 series, the network was improved, the shallow feature detection layer was added, the three-scale feature detection was changed to four-scale feature detection, and the up sampling was increased by four times. Add the CBAM attention module; Use lightweight GhostBottleneck instead of Bottleneck structures. The results in our experiments show that YOLOv5s-FCG raises the average detection accuracy (mAP) by 2.0% compared with YOLOv5s on the ourselves’ riding safety helmet data set and 1.5% on the NWPU-VHR-10 public data set. The proposed algorithm not only ensured the detection rate, volume, computation and number of parameters, but also improved the detection accuracy. And it had good adaptability and generalization ability in complex road environments such as poor light and small targets.