School of Information Science and Engineering, Zhejiang Sci-Tech University, Hangzhou 310018, China.
School of Aeronautics and Astronautics, Zhejiang University, Hangzhou 310027, China.
Sensors (Basel). 2022 Dec 14;22(24):9843. doi: 10.3390/s22249843.
At many construction sites, whether to wear a helmet is directly related to the safety of the workers. Therefore, the detection of helmet use has become a crucial monitoring tool for construction safety. However, most of the current helmet wearing detection algorithms are only dedicated to distinguishing pedestrians who wear helmets from those who do not. In order to further enrich the detection in construction scenes, this paper builds a dataset with six cases: not wearing a helmet, wearing a helmet, just wearing a hat, having a helmet, but not wearing it, wearing a helmet correctly, and wearing a helmet without wearing the chin strap. On this basis, this paper proposes a practical algorithm for detecting helmet wearing states based on the improved YOLOv5s algorithm. Firstly, according to the characteristics of the label of the dataset constructed by us, the K-means method is used to redesign the size of the prior box and match it to the corresponding feature layer to increase the accuracy of the feature extraction of the model; secondly, an additional layer is added to the algorithm to improve the ability of the model to recognize small targets; finally, the attention mechanism is introduced in the algorithm, and the CIOU_Loss function in the YOLOv5 method is replaced by the EIOU_Loss function. The experimental results indicate that the improved algorithm is more accurate than the original YOLOv5s algorithm. In addition, the finer classification also significantly enhances the detection performance of the model.
在许多建筑工地,工人是否戴安全帽直接关系到他们的安全。因此,检测安全帽的佩戴情况已成为建筑安全的重要监控手段。然而,目前大多数的安全帽佩戴检测算法仅专注于区分戴安全帽的行人和不戴安全帽的行人。为了进一步丰富施工现场的检测,本文构建了一个包含六个案例的数据集:未戴安全帽、戴安全帽、仅戴帽子、戴了安全帽但未系好、正确戴了安全帽、未系好安全帽下颚带。在此基础上,本文提出了一种基于改进的 YOLOv5s 算法的实用的安全帽佩戴状态检测算法。首先,根据我们构建的数据集标签的特点,使用 K-means 方法重新设计先验框的大小,并将其与相应的特征层匹配,以提高模型特征提取的准确性;其次,在算法中添加一个额外的层,以提高模型识别小目标的能力;最后,在算法中引入注意力机制,并将 YOLOv5 方法中的 CIOU_Loss 函数替换为 EIOU_Loss 函数。实验结果表明,改进后的算法比原始的 YOLOv5s 算法更准确。此外,更精细的分类也显著提高了模型的检测性能。
Sensors (Basel). 2022-12-14
Sensors (Basel). 2023-2-3
Sensors (Basel). 2023-6-22
Sci Rep. 2024-3-25
Math Biosci Eng. 2023-3-6
Sensors (Basel). 2022-9-5
Sci Rep. 2023-12-5
Sci Rep. 2024-7-30
Sensors (Basel). 2022-8-13
Sensors (Basel). 2024-10-30
Sci Rep. 2024-7-30