Author:
Wang Pengfei,Huang Hanming,Wang Mengqi,Li Bingjun
Abstract
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.
Subject
General Physics and Astronomy
Cited by
8 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献