Suppr超能文献

基于深度学习的头盔佩戴检测方法研究。

Research on helmet wearing detection method based on deep learning.

机构信息

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.

Abstract

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%,并且模型的检测速度没有太大下降,更适合实时安全帽目标检测,满足了各种工作场景下的头盔检测要求。

https://cdn.ncbi.nlm.nih.gov/pmc/blobs/2a6f/10963364/5638521be5bb/41598_2024_57433_Fig1_HTML.jpg

文献检索

告别复杂PubMed语法,用中文像聊天一样搜索,搜遍4000万医学文献。AI智能推荐,让科研检索更轻松。

立即免费搜索

文件翻译

保留排版,准确专业,支持PDF/Word/PPT等文件格式,支持 12+语言互译。

免费翻译文档

深度研究

AI帮你快速写综述,25分钟生成高质量综述,智能提取关键信息,辅助科研写作。

立即免费体验