基于子特征融合的轻量化安全帽检测方法技术

本发明专利技术涉及一种基于子特征融合的轻量化安全帽检测方法,包括以下步骤：步骤S1:获取安全帽数据图片，并预处理，构建安全帽检测数据集；步骤S2：构建子特征融合的轻量化网络模块，通过低代价卷积操作生成子特征图并与初始特征图进行融合；步骤S3：构建轻量化目标检测网络；步骤S4：对YOLOv5目标检测算法的训练超参数进行调优，利用安全帽检测数据集训练轻量化目标检测网络得到安全帽检测模型；步骤S5：根据安全帽检测模型对输入图片进行检测，获取初步检测结果，对初步检测结果进行解码后采用改进的非极大值抑制算法筛选出最终的检测结果，将检测结果绘制在原图中。本发明专利技术能够有效减少检测模型的模型大小，并且对工人是否佩戴安全帽进行精准检测。帽进行精准检测。帽进行精准检测。

【技术实现步骤摘要】
基于子特征融合的轻量化安全帽检测方法


[0001]本专利技术涉及图像识别与计算机视觉领域，具体涉及一种基于子特征融合的轻量化安全帽检测方法。

技术介绍

[0002]在建筑工地中，施工人员必须佩戴安全帽保障自身的工作安全性。建筑业是所有行业中较为危险的行业之一，美国劳工统计局(BLS) 的数据显示，建筑业的死亡人数远高于其他所有行业，致命伤害率也高于全部行业的全国平均水平。负责人员监管不力，施工人员没有严格遵守建筑工地的相关安全规定，佩戴安全帽和安全背心等安全防护设备，导致安全事故频繁发生。美国职业安全与健康管理局(OSHA)要求所有在施工现场的工作人员佩戴安全帽，以最大限度地减少安全事故的发生或降低安全事故导致的伤害。美国国家职业安全与健康研究所(NIOSH)的一份报告显示，在2003年至2010年期间，全美共有2,210人因脑外伤(TBI)死亡，占同期建筑业所有死亡人数的25％。在工业生产过程中，脑外伤事故发生的最常见原因是工人从高处坠落或物体掉落砸伤工人头部，佩戴安全帽可以最大程度地减少头部受到的伤害。研究表明，如果工人在工作时佩戴安全帽，可以避免大多数安全事故的发生。
[0003]目前，对于以上安全帽佩戴情况的识别主要以人工检查的方式为主。人工检查容易被各种各样的因素干扰降低效率，并且无法实行全天候监控且浪费人力资源，所以这种方法效率比较低下，各安全监管部门的实际需求无法得到充分满足。通过监控摄像头等设备，并利用基于机器视觉的检测算法对工人的安全帽佩戴情况进行自动识别，可以实现实时监控，避免人工检测中由于各种主观因素造成的漏检和误检等问题，在节省人力资源的同时还可以提高监管的效率及自动化水平。由于深度学习模型的体积和计算量过于庞大，难以直接在监控摄像头、无人机等互联网边缘设备上使用，且现有的大多数检测模型检测速度较慢，无法满足实时检测需求。

技术实现思路

[0004]有鉴于此，本专利技术的目的在于提供一种基于子特征融合的轻量化安全帽检测方法,能够有效减少检测模型的模型大小，并且对工人是否佩戴安全帽进行精准检测。
[0005]为实现上述目的，本专利技术采用如下技术方案：
[0006]一种基于子特征融合的轻量化安全帽检测方法,包括以下步骤：
[0007]步骤S1:获取安全帽数据图片，并预处理，构建安全帽检测数据集；
[0008]步骤S2：构建子特征融合的轻量化网络模块，通过低代价卷积操作生成子特征图并与初始特征图进行融合；
[0009]步骤S3：利子特征融合的轻量化网络模块和STEM模块构建轻量化Backbone网络，利用普通卷积模块和C3模块构建Neck和Head，再将轻量化Backbone网络，Neck和Head组成一个轻量化目标检测网络；
[0010]步骤S4：对YOLOv5目标检测算法的训练超参数进行调优，利用安全帽检测数据集训练轻量化目标检测网络得到安全帽检测模型；
[0011]步骤S5：根据安全帽检测模型对输入图片进行检测，获取初步检测结果，对初步检测结果进行解码后采用改进的非极大值抑制算法筛选出最终的检测结果，将检测结果绘制在原图中。
[0012]进一步的，所述步骤S1，具体为：
[0013]步骤S11；获取安全帽相关的数据图片，筛选图片并统一命名；
[0014]步骤S12：采用邻域去噪、中值滤波对图片进行处理；
[0015]步骤S13：确定安全帽图片中的物体类别，使用LabelImg标注工具对预处理后的数据图片进行标注，得到并保存标注信息；
[0016]步骤S14：使用几何变换对标注后的数据图片进行数据增强，然后根据YOLOv5模型要求制作数据集，将所有数据按比例分为训练集，验证集和测试集，根据包含图片数据标注信息的xml文件生成训练模型所需的txt文件。
[0017]进一步的，所述子特征融合的轻量化网络模块包括第一卷积层、深度可分离卷积层和第二卷积层；
[0018]所述第一卷积层，实际输出通道数设为传入输出通道参数值的一半，对输入特征图进行卷积操作后生成输出特征图；
[0019]所述第二卷积层，同样将该卷积层的实际输出通道数设为传入输出通道参数值的一半，该卷积层将第一卷积层的输出特征图作为输入特征图，进行卷积之后生成子特征图作为输出特征图；
[0020]将第一个卷积层的输出特征图和第二个卷积层生成的子特征图进行融合，作为最终的输出特征图。
[0021]进一步的，所述步骤S3，具体为：
[0022]步骤S31：利用子特征融合模块和STEM模块构建轻量化 Backbone网络，Backbone网络的第一层使用STEM模块；
[0023]步骤S32：利用普通卷积模块和C3模块构建Neck和Head，普通卷积模块和C3模块交叉搭建构成Neck和Head网络；
[0024]步骤S33：将Backbone，Neck和Head网络组成一个轻量化目标检测网络，其中Backbone和Neck之间进行多尺度特征融合，提高检测网络精度。
[0025]进一步的，所述C3模块由三个卷积层和一个标准瓶颈层组成，
[0026]进一步的，所述步骤S4，具体为：
[0027]步骤S41：根据预设实验得出超参数的最优值，调优超参数将使训练模型达到最优；
[0028]步骤S42：将标注文件中的标注框参数作为输入数据，使用k
‑
means聚类算法根据标注框计算数据集的先验框anchor；
[0029]步骤S43：设置训练迭代次数为N，设置数据图片读取批次 batch
‑
sizes，训练完成后得到安全帽检测模型。
[0030]进一步的，所述步骤S42具体为：
[0031]步骤S421：使用数据集图像的宽高w
img
,h
img
对标注框boundingbox 的宽高w
box
,h
box
做归一化，具体为：
[0032][0033]其中，W
Normalize
，H
Normalize
为归一化后的宽和高；
[0034]步骤S422：令anchor＝(w
anchor
,h
anchor
),box＝(w
box
,h
box
)，w
anchor
,h
anchor
为先验框anchor的宽和高，使用交并比IOU作为度量，其计算方式如下：
[0035][0036]IOU的取值在0到1之间，两个box重合度越高则IOU值越大， d为最终度量，其计算公式为：
[0037]d＝1
‑
IOU(box,anchor)
[0038]步骤S423：在数据集中随机选取box_k个bounding box作为初始先验框anchor，然后计算每个标注框和这些anchor的IOU度本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于子特征融合的轻量化安全帽检测方法,其特征在于，包括以下步骤：步骤S1:获取安全帽数据图片，并预处理，构建安全帽检测数据集；步骤S2：构建子特征融合的轻量化网络模块，通过低代价卷积操作生成子特征图并与初始特征图进行融合；步骤S3：利子特征融合的轻量化网络模块和STEM模块构建轻量化Backbone网络，利用普通卷积模块和C3模块构建Neck和Head，再将轻量化Backbone网络，Neck和Head组成一个轻量化目标检测网络；步骤S4：对YOLOv5目标检测算法的训练超参数进行调优，利用安全帽检测数据集训练轻量化目标检测网络得到安全帽检测模型；步骤S5：根据安全帽检测模型对输入图片进行检测，获取初步检测结果，对初步检测结果进行解码后采用改进的非极大值抑制算法筛选出最终的检测结果，将检测结果绘制在原图中。2.根据权利要求1所述的基于子特征融合的轻量化安全帽检测方法，其特征在于，所述步骤S1，具体为：步骤S11；获取安全帽相关的数据图片，筛选图片并统一命名；步骤S12：采用邻域去噪、中值滤波对图片进行处理；步骤S13：确定安全帽图片中的物体类别，使用LabelImg标注工具对预处理后的数据图片进行标注，得到并保存标注信息；步骤S14：使用几何变换对标注后的数据图片进行数据增强，然后根据YOLOv5模型要求制作数据集，将所有数据按比例分为训练集，验证集和测试集，根据包含图片数据标注信息的xml文件生成训练模型所需的txt文件。3.根据权利要求1所述的基于子特征融合的轻量化安全帽检测方法，其特征在于，所述子特征融合的轻量化网络模块包括第一卷积层、深度可分离卷积层和第二卷积层，具体为；所述第一卷积层，实际输出通道数设为传入输出通道参数值的一半，对输入特征图进行卷积操作后生成输出特征图；所述第二卷积层，同样将该卷积层的实际输出通道数设为传入输出通道参数值的一半，该卷积层将第一卷积层的输出特征图作为输入特征图，进行卷积之后生成子特征图作为输出特征图；将第一个卷积层的输出特征图和第二个卷积层生成的子特征图进行融合，作为最终的输出特征图。4.根据权利要求1所述的基于子特征融合的轻量化安全帽检测方法，其特征在于，所述步骤S3，具体为：步骤S31：利用子特征融合模块和STEM模块构建轻量化Backbone网络，Backbone网络的第一层使用STEM模块；步骤S32：利用普通卷积模块和C3模块构建Neck和Head，普通卷积模块和C3模块交叉搭建构成Neck和Head网络；步骤S33：将Backbone，Neck和Head网络组成一个轻量化目标检测网络，其中Backbone和Neck之间进行多尺度特征融合，提高检测网络精度。5.根据权利要求4所述的基于子特征融合的轻量化安全帽检测方法，其特征在于，所述
C3模块由三个卷积层和一个标准瓶颈层组成。6.根据权利要求1所述的基于子特征融合的轻量化安全帽检测方法，其特征在于，所述步骤S4，具体为：步骤S41：根据预设实验得出超参数的最优值，调优超参数将使训练模型达到最优；步骤S42：将标注文件中的标注框参数作为输入数据，使用k
‑
means聚类算法根据标注框计算数据集的先验框anchor；步骤S43：设置训练迭代次数为N，设置数据图片读取批次batch
‑
sizes，训练完成后得到安全帽检测模型。7.根据权利要求6所述的基于子特征融合的轻量化安全帽检测方法，其特征在于，所述步骤S42具体为：步骤S421：使用数据集图像的宽高w
img
,h
img
对标注框bounding box的宽高w...

【专利技术属性】
技术研发人员：柯逍，陈文垚，
申请(专利权)人：福州大学，
类型：发明
国别省市：