基于改进YOLOv5的化工生产过程中安全防护品佩戴检测方法技术

本发明专利技术公开了一种基于改进YOLOv5的化工安全防护品佩戴检测方法，获取训练图像，使用MixUp算法对安全帽图像数据和口罩图像数据进行加权融合，得到更多存在双目标的训练图像；搭建包含注意力机制NAM模块和Encoder结构的改进的YOLOv5网络模型，在骨干网络中插入注意力机制NAM模型，在颈部模块添加了改进的Encoder结构；将训练数据集和测试数据集输入到改进的YOLOv5网络模型中，进行优化训练，得到改进的安全防护品佩戴检测模型；将待检测图像数据集输入上述模型，得到待检测图像中的安全防护品的检测结果。与现有技术相比，本发明专利技术充分优化了网络模型的特征提取能力，有效地提高了小目标检测的精确度，提高了化工生产过程中安全防护品佩戴检测的准确性。中安全防护品佩戴检测的准确性。中安全防护品佩戴检测的准确性。

【技术实现步骤摘要】
基于改进YOLOv5的化工生产过程中安全防护品佩戴检测方法


[0001]本专利技术涉及化工生产场景下安全防护品佩戴检测与机器视觉的
，具体涉及一种基于改进YOLOv5的化工生产过程中安全防护品佩戴检测方法。

技术介绍

[0002]随着化工行业的发展，化工安全越来越被重视和关注。安全帽和口罩作为一种对人员头部防护和肺部防护起着关键作用的设备，在化工生产过程中，佩戴安全防护品的监管十分必要，研究一种精度高、鲁棒性强的安全防护品佩戴检测算法可以保障生产人员的人身安全，降低安全事故的危害，具有重要的实际意义。
[0003]安全防护品佩戴检测的研究可以分成两大类∶基于传感器的检测方法和基于计算机视觉的检测方法。基于传感器的检测方法通过在安全帽上嵌入各类传感器，如压力传感器、电子标签等实现与外界通信。但这就会对工人的正常作业造成干扰，跟踪设备通常都会有距离限制，且成本较高。
[0004]基于计算机视觉的检测方法主要包括基于深度学习的两阶段检测算法和基于深度学习的单阶段检测算法。由于两阶段安全帽检测算法需要预先生成候选框，再对候选框进行分类和回归，边界框经过两次微调，检测准确度比单阶段安全防护品佩戴检测算法要高，但损失了检测速度。这就导致两阶段安全防护品佩戴检测算法不能够适用于对速度要求极高的检测任务。
[0005]基于深度学习的单阶段安全防护品检测算法模型结构简单且检测速度快，能够很好的应用于安全防护品佩戴检测任务中，但单阶段检测算法的检测准确度有所下降。YOLOv5沿用了YOLOv4的网络结构，采用了Mosaic数据增强方法，采用自适应锚框，检测速度很快，集成了YOLOv3和YOLOv4的部分特性，检测速度远超YOLOv4，但是对于检测复杂场景中和密集小目标中作业人员安全防护品佩戴情况的检测准确度依旧有所欠缺。

技术实现思路

[0006]专利技术目的：针对现有技术中存在的问题，本专利技术提出一种基于改进YOLOv5的化工安全防护品佩戴检测方法，能够在小目标的情况下，精确的识别是否佩戴安全防护品。
[0007]技术方案：本专利技术提出一种基于改进YOLOv5的化工安全防护品佩戴检测方法，包括以下步骤：
[0008]步骤1：获取训练图像并进行预处理，使用MixUp数据增强算法对安全帽图像数据和口罩图像数据进行加权融合，得到更多存在双目标的训练图像，按比例分配得到训练数据集D1和测试数据集D2；
[0009]步骤2：搭建包含注意力机制NAM模块和Encoder结构的改进的YOLOv5网络模型；
[0010]步骤3：搭将训练数据集D1和测试数据集D2输入改进的YOLOv5网络模型中，对网络进行训练与测试，得到改进的安全防护品佩戴检测模型Mod；
[0011]步骤4：搭将待检测图像输入模型Mod中，得到待检测图像中的安全防护品的检测
结果。
[0012]进一步地，所述步骤1中获取训练图像并进行预处理具体包括以下步骤：
[0013]步骤1.1：通过化工厂的监控图像和公共数据集获取安全防护品佩戴情况的图像，其中安全防护品包括安全帽和口罩；
[0014]步骤1.2：对得到的图像进行筛选，选取安全防护品佩戴图像，包括不同场景、光线、遮挡和多目标情况下的图片，并且删除错误图片；
[0015]步骤1.3：对筛选后的图像进行归一化处理，将图像按照统一规定命名，统一图像为RGB三通道彩色图，统一图像尺寸；
[0016]步骤1.4：利用LabelImg工具对图像进行手动标注，标注图像中安全帽和口罩的位置和类别，并生成XML标签文件。
[0017]进一步地，所述步骤1中使用MixUp算法对安全帽图像数据和口罩图像数据进行加权融合具体包括：
[0018]步骤1.5：通过MixUp数据增强算法随机对一张佩戴口罩的图像和一张佩戴安全帽的图像初始图像进行加权处理，得到更多的同时存在安全帽和口罩的图像数据；
[0019]步骤1.6：加权融合公式如下：
[0020][0021]其中，表示新生成的同时存在安全帽和口罩类别的图像数据，x
a
表示随机一张佩戴口罩的图像，x
b
表示随机一张佩戴安全帽的图像，λ是MixUp方法的权值，取值范围为λ∈[0.4，0.6]；
[0022]步骤1.7：将数据增强后的数据集按照8∶2的比例划分，得到训练数据集D1和测试数据集D2。
[0023]进一步地，所述步骤2改进的YOLOv5网络模型包括以下结构：
[0024]改进的YOLOv5网络主要包括改进的Backbone模块、改进的Neck模块和Head模块；
[0025]所述改进的Backbone模块包括Focus模块、CSP模块、SPP模块和注意力机制NAM模块；Focus模块首先把输入的安全防护品佩戴图片进行切片处理，然后进行Concat操作，再对切片后的图像进行卷积操作，得到304
×
304大小的初始特征图，CSP模块由卷积块和若干个残差结构组成，对网络中的梯度信息进行优化，通过在特征图中融合梯度信息降低网络的参数量，SPP模块用于特征提取，使用三个不同卷积核对CSP中输出的特征图进行下采样，得到三个特征图，再进行Concat操作，最后将融合后的特征图进行卷积操作；在CSP模块和SPP模块后均插入注意力机制NAM模块，通过使用BN的缩放因子来表示权值的重要程度，NAM模块主要包括通道注意力模块和空间注意力模块；
[0026]改进的Neck模块主要包括FPN模块和PAN模块，FPN模块自上而下的对特征图进行上采样操作，增大特征图，将得到的特征图与CSP模块中输出的特征图进行Concat操作，PAN模块自下而上的对特征图进行下采样操作，缩小特征图，将得到的特征图与FPN模块中输出的特征图进行Concat操作，在每个输出后插入改进的Encoder结构，Encoder结构的输出特征图和FPN模块的输出特征图进行Concat操作，进一步的提取更多的特征信息；
[0027]改进的Encoder结构，首先对输入的特征图进行展平操作，将特征图展平为一个序列，对该序列进行Positional Encoding操作，用于对图像区域进行位置编码，得到位置序
列X，接着是一个Multi
‑
Head Attention层，将Positional Encoding操作的输出和Multi
‑
Head Attention层的输出进行Add操作，再对输出的特征矩阵进行归一化处理并输入多层感知器中，将多层感知器的输出和Add操作的输出再一次进行Add操作，最后对输出的特征矩阵进行归一化处理和Rearrange操作，使改进的Encoder结构的输出符合Head模块的输入要求；
[0028]将上述修改后的模块和结构按照YOLOv5的网络形式进行堆叠，得到包含注意力机制NAM模块和Encoder结构的改进的YOLOv5网络模型。
[0029]进一步地本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于改进YOLOv5的化工安全防护品佩戴检测方法，其特征在于，包括以下步骤：步骤1：获取训练图像并进行预处理，使用MixUp数据增强算法对安全帽图像数据和口罩图像数据进行加权融合，得到更多存在双目标的训练图像，按比例分配得到训练数据集D1和测试数据集D2；步骤2：搭建包含注意力机制NAM模块和Encoder结构的改进的YOLOv5网络模型；步骤3：搭将训练数据集D1和测试数据集D2输入改进的YOLOv5网络模型中，对网络进行训练与测试，得到改进的安全防护品佩戴检测模型Mod；步骤4：搭将待检测图像输入模型Mod中，得到待检测图像中的安全防护品的检测结果。2.根据权利要求1所述的基于改进YOLOv5的化工安全防护品佩戴检测方法，其特征在于，所述步骤1中获取训练图像并进行预处理具体包括以下步骤：步骤1.1：通过化工厂的监控图像和公共数据集获取安全防护品佩戴情况的图像，其中安全防护品包括安全帽和口罩；步骤1.2：对得到的图像进行筛选，选取安全防护品佩戴图像，包括不同场景、光线、遮挡和多目标情况下的图片，并且删除错误图片；步骤1.3：对筛选后的图像进行归一化处理，将图像按照统一规定命名，统一图像为RGB三通道彩色图，统一图像尺寸；步骤1.4：利用LabelImg工具对图像进行手动标注，标注图像中安全帽和口罩的位置和类别，并生成XML标签文件。3.根据权利要求2所述的基于改进YOLOv5的化工安全防护品佩戴检测方法，其特征在于，所述步骤1中使用MixUp算法对安全帽图像数据和口罩图像数据进行加权融合具体包括：步骤1.5：通过MixUp数据增强算法随机对一张佩戴口罩的图像和一张佩戴安全帽的图像初始图像进行加权处理，得到更多的同时存在安全帽和口罩的图像数据；步骤1.6：加权融合公式如下：其中，表示新生成的同时存在安全帽和口罩类别的图像数据，x
a
表示随机一张佩戴口罩的图像，x
b
表示随机一张佩戴安全帽的图像，λ是MixUp方法的权值，取值范围为λ∈[0.4，0.6]；步骤1.7：将数据增强后的数据集按照8∶2的比例划分，得到训练数据集D1和测试数据集D2。4.根据权利要求1所述的基于改进YOLOv5的化工安全防护品佩戴检测方法，其特征在于，所述步骤2改进的YOLOv5网络模型包括以下结构：改进的YOLOv5网络主要包括改进的Backbone模块、改进的Neck模块和Head模块；所述改进的Backbone模块包括Focus模块、CSP模块、SPP模块和注意力机制NAM模块；Focus模块首先把输入的安全防护品佩戴图片进行切片处理，然后进行Concat操作，再对切片后的图像进行卷积操作，得到304
×
304大小的初始特征图，CSP模块由卷积块和若干个残差结构组成，对网络中的梯度信息进行优化，通过在特征图中融合梯度信息降低网络的参数量，SPP模块用于特征提取，使用三个不同卷积核对CSP中输出的特征图进行下采样，得到三个特征图，再进行Concat操作，最后将融合后的特征图进行卷积操作；在CSP模块和SPP模
块后均插入注意力机制NAM模块，通过使用BN的缩放因子来表示权值的重要程度，NAM模块主要包括通道注意力模块和空间注意力模块；改进的Neck模块主要包括FPN模块和PAN模块，FPN模块自上而下的对特征图进行上采样操作，增大特征图，将得到的特征图与CSP模块中输出的特征图进行Concat操作，PAN模块自下而上的对特征图进行下采样操作，缩小特征图，将得到的特征图与FPN模块中输出的特征图进行Concat操作，在每个输出后插入改进的Encoder结构，Encoder结构的输出特征图和FPN模块的输出特征图进行Concat操作，进一步的提取更多的特征信息；改进的Encoder结构，首先对输入的特征图进行展平操作，将特征图展平为一个序列，对该序列进行Positional Encoding操作，用于对图像区域进行位置编码，得到位置序列X，接着是一个Multi
‑
Head Attention层，将Positional Encoding操作的输出和Multi
‑
Head Attention层的输出进行Add操作，再对输出...

【专利技术属性】
技术研发人员：陈晓兵，张冰莹，康丽，包涵，张润，周冬冬，郭舒心，
申请(专利权)人：淮阴工学院，
类型：发明
国别省市：