Detection Algorithm for Safety Helmet Wearing of Chemical Plant Personnel Based on Improved YOLOv5m
Qiangqiang Li, Kun Yu, Haoran Wang, Qianjun Guan, Shihong Gao, Jiamin Jiang · 2023
Wearing a safety helmet at the work site of a chemical plant can effectively prevent safety accidents caused by head injuries, so it is very important to detect whether employees wear safety helmets. In order to solve problems such as serious information loss, large number of parameters, and weak detection ability of small targets in complex scenes in the sampling process of helmet detection algorithm. In this paper, an improved lightweight safety helmet detection algorithm LP- YOLOv5m (Lightweight And High Precision YOLOv5m) based on YOLOv5m is designed. LP-YOLOv5m incorporates the Wise-IoU (Wiou) loss function to enhance model generalization. Additionally, we add a small object detection layer to improve the feature extraction ability for small objects. Furthermore, upsampling was performed using CARAFE. Finally, a lightweight module named C3-F was proposed, which replaced the Bottleneck in the C3 module with FasterNet Block, which could effectively reduce the network model size. Comparative laboratory findings demonstrate that our improved model achieves a mAP50 of 92.1%, surpassing YOLOv5m by 3 percentage points while reducing the model size by 23.3%. The LP-YOLOv5m model can be better applied to practical scene requirements, and can effectively solve practical problems.