School of Artificial Intelligence and Big Data, Hulunbeier University, Inner Mongolia, 021008, Hailar, China.
Information Science and Engineering School, Northeastern University, Shenyang, 110004, China.
Sci Rep. 2024 Mar 25;14(1):7010. doi: 10.1038/s41598-024-57433-z.
The vigorous development of the construction industry has also brought unprecedented safety risks. The wearing of safety helmets at the construction site can effectively reduce casualties. As a result, this paper suggests employing a deep learning-based approach for the real-time detection of safety helmet usage among construction workers. Based on the selected YOLOv5s network through experiments, this paper analyzes its training results. Considering its poor detection effect on small objects and occluded objects. Therefore, multiple attention mechanisms are used to improve the YOLOv5s network, the feature pyramid network is improved into a BiFPN bidirectional feature pyramid network, and the post-processing method NMS is improved into Soft-NMS. Based on the above-improved method, the loss function is improved to enhance the convergence speed of the model and improve the detection speed. We propose a network model called BiFEL-YOLOv5s, which combines the BiFPN network and Focal-EIoU Loss to improve YOLOv5s. The average precision of the model is increased by 0.9% the recall rate is increased by 2.8%, and the detection speed of the model does not decrease too much. It is better suited for real-time safety helmet object detection, addressing the requirements of helmet detection across various work scenarios.
建筑行业的蓬勃发展也带来了前所未有的安全风险。施工现场佩戴安全帽可以有效减少人员伤亡。因此,本文建议采用基于深度学习的方法实时检测建筑工人佩戴安全帽的情况。本文通过实验选择了 YOLOv5s 网络,并对其训练结果进行了分析。考虑到其对小目标和遮挡目标的检测效果较差,因此使用了多种注意力机制对 YOLOv5s 网络进行改进,将特征金字塔网络改进为 BiFPN 双向特征金字塔网络,并将后处理方法 NMS 改进为 Soft-NMS。基于上述改进方法,改进损失函数以增强模型的收敛速度并提高检测速度。我们提出了一种名为 BiFEL-YOLOv5s 的网络模型,该模型结合了 BiFPN 网络和 Focal-EIoU Loss 来改进 YOLOv5s。模型的平均精度提高了 0.9%,召回率提高了 2.8%,并且模型的检测速度没有太大下降,更适合实时安全帽目标检测,满足了各种工作场景下的头盔检测要求。
Sci Rep. 2024-3-25
Sensors (Basel). 2023-2-3
Sensors (Basel). 2023-6-22
Sensors (Basel). 2022-12-14
Sci Rep. 2023-10-23
Sci Rep. 2024-7-30
Front Plant Sci. 2024-1-11
Sci Rep. 2023-12-5
PeerJ Comput Sci. 2020-12-7
Sensors (Basel). 2022-3-17
Sensors (Basel). 2023-2-6
Sensors (Basel). 2023-1-21
IEEE Trans Pattern Anal Mach Intell. 2016-6-6