[发明专利]基于改进YOLOV5模型的工地安全帽佩戴检测方法

说明书

本发明公开了一种基于改进YOLOV5模型的工地安全帽佩戴检测方法，基于单阶段原始的YOLOV5目标检测模型，通过优化网络结构降低最小感受野，利用池化金字塔及注意力机制等手段规范输入尺寸，模糊背景差异，增强小目标学习能力，本发明所提出的方法在基准网络基础上只增加了少量参数及计算开销，并显著提升了小目标检测效果，克服原始YOLOV5目标检测模型小目标场景检测性能较低的缺点，改进后的YOLOV5模型能够在施工场景下的安全帽检测任务中有着优异的检测表现，在同一场景下比原始YOLOV5网络检测到更多正样本，提高了施工场景下安全帽佩戴检测的可靠性，降低漏检、误检的情况发生。

技术领域

本发明属于建筑行业智能化处理技术领域，尤其涉及基于改进YOLOV5模型的工地安全帽佩戴检测方法。

背景技术

建筑业是高风险、监管薄弱、信息化程度较低的传统行业，在复杂的施工环境下，工人的不安全行为很容易导致安全事故，威胁施工人员生命安全。据统计，在工地安全事故中，头部周边损伤占大部分，减少该类事故受伤的有效措施在于正确使用安全防护设备。由于安全帽具有多种防护能力，如电绝缘性、耐穿刺、抗冲击性等优点，可以有效减少头部和颈部损伤。因此，运用智能技术实现安全帽佩戴的自动化检测，是保障施工人员生命财产的必要手段。

近年来，随着计算机视觉技术的不断发展，目标检测领域在工业生产中具有广泛应用。早期的安全帽佩戴检测方法主要有Hough、HOG等，通过对行人进行检测后，利用颜色、形状等特征对安全帽进行再识别。随着卷积神经网络的发展，目标检测算法也转向了基于深度神经网络的检测算法。目前基于深度神经网络的目标检测算法可以分为单阶段目标检测模型和双阶段目标检测模型两类。单阶段目标检测模型主要以SSD和YOLO为主；而双阶段目标检测模型主要有RCNN、FASTRCNN等。

现有原始的YOLOV5目标检测模型相比于之前旧版本权衡了速度与检测精度，有着计算开销小、精度高的优异性能，并且YOLOV5s模型尺寸更小，利于快速部署。但现有原始的YOLOV5目标检测模型设计时只是考虑了通用性，却难以满足小目标、密集场景等视觉任务。尤其在进行安全帽检测时，拍摄场景的远近和复杂的施工环境会产生大量的小目标，造成较多的漏检、误检行为。

发明内容

本发明的目的在于克服现有原始的YOLOV5目标检测模型针对小目标检测场景的不足，提供一种基于改进YOLOV5模型的工地安全帽佩戴检测方法，降低小目标检测时的漏检、误检情况，提高施工场景下小目标检测能力。

本发明的目的是通过以下技术方案来实现的：

基于改进YOLOV5模型的工地安全帽佩戴检测方法，其特殊之处在于，包括：

步骤1、收集并划分数据集：

收集开源数据集SHWD、SHD及其他网络图片，制作所需数据集，将数据集划分为训练集、验证集和测试集。

步骤2、对原始的YOLOV5目标检测模型进行改进：

通过在原始的YOLOV5目标检测模型基础上优化网路结构，得到改进后的YOLOV5模型；所述改进后的YOLOV5模型采用K-Means++聚类算法实时计算锚框，加速模型收敛；采用Swin Tranformer Block作为Backbone高层语义提取网络，增强模型上下文特征提取能力；添加小目标检测层，降低网络最小感受野；特征融合部分融入SPPF特征融合金字塔，通过多层次的空间箱从不同的角度进行特征提取再聚合，提升算法对物体变形的鲁棒性；采用Bottleneck Transformer优化特种融合部分的卷积层，增强网络特征融合能力；采用Shuffle Attention作为注意力机制更新特征矩阵，增强小目标特征表达能力。

步骤3、利用改进后的YOLOV5模型进行训练：

将所述步骤1数据集中的训练集送入改进后的YOLOV5模型进行训练，并保存训练过程中改进后的YOLOV5模型在验证集上检测准确率最高的权重参数，并将权重文件命名为best.pt。