[发明专利]一种基于神经网络目标检测的厂区工装检测方法及系统

说明书

本申请实施例提供的一种基于神经网络目标检测的厂区工装检测方法及系统，包含了对数据集进行划分、损失函数与目标函数的设计、目标检测模型的训练与推理等。此处使用的目标检测模型主要采用YOLOv6模型，也即YOLO架构的目标检测框架。针对工作服样本图像的训练样本数据集进行划分，主要采用二分类来进行区分判断是否有穿戴工装，将数据集传入目标检测框架中生成相应的检测模型，最终即可得到推理检测的结果。

技术领域

本申请涉及工业视觉技术领域，具体涉及一种基于神经网络目标检测的厂区工装检测方法及系统。

背景技术

近年来，由于操作不规范和着装不规范的原因，已经成为作业人员在工业厂区发生意外的主要原因之一。标准化作业规范检测也随之成为工业检测的一个技术指标。

普通衣服在天气干燥或者作业时衣服之间的摩擦而产生静电，但是在变电站、油库等特殊场合，绝对不能出现静电，所以都必须穿着特制的防静电工作服，面料必须是防静电面料才可以。而且为了方便管理，对内部工作人员、司机、施工人员等关键人员统一服装款式。为保证内部所有人员穿戴了防静电工作服，并且工作人员穿戴了统一的服装，对内部的所有进出人员进行工作服的检测。然而，仅依靠人力进行监督查看，需耗费大量的人力资源。

鉴于此，结合深度学习与目标检测对工业厂区的工服工装进行识别检测具有重大意义，本申请提出一种基于神经网络目标检测的厂区工装检测方法及系统，能够精准、快速地对厂区工装进行检测，识别出工作场所中工作人员是否按要求穿戴指定工作服。

发明内容

本申请实施例提出了一种基于神经网络目标检测的厂区工装检测方法及系统，来解决以上背景技术部分提到的技术问题。

第一方面，本申请实施例提供了一种基于神经网络目标检测的厂区工装检测方法，包括以下步骤：

S1、获取多张工作服样本图像，对所述工作服样本图像标注标签，将全部的所述工作服样本图像及其对应的标签确定为训练样本数据集；

S2、将所述训练样本数据集按比例划分为训练集、验证集和测试集；

S3、构建目标检测模型：将所述训练样本数据集中的图像输入到backbone网络中，根据所述backbone网络中的三层输出，在neck层通过Rep-PAN网络继续输出三层不同尺寸大小的特征图，将所述特征图输入到head层中，并且对所述特征图进行三类任务预测；构建所述目标检测模型的损失函数；

S4、将所述训练集输入到构建好的所述目标检测模型中进行训练，所述损失函数不断迭代至收敛，获得最优网络权重，所述目标检测模型通过所述验证集进行预测，并通过所述测试集进行测试验证；以及

S5、设置固定阈值，并根据所述固定阈值输出目标检测结果。

通过上述技术方案，根据指定的工作服训练目标检测模型，大量采集工作服样本，通过深度学习，识别出工作场所中工作人员是否按要求穿戴指定工作服，对未穿指定工作服的人员，进行抓图报警。在实际应用中，还可以进行语音提示报警。

在具体的实施例中，在步骤S3中，将所述训练样本数据集中的图像输入到backbone网络中，包括以下子步骤：

S311、输入640*640*3的所述训练样本数据集中的图像到所述backbone网络中，通过stem层输出为320*320*3*2的图像；

S312、所述stem层连接多个ERBlock，每个所述ERBlock均做特征层的下采样和channel的增大，每个所述ERBlock由一个RVB和一个RB组成，在所述RVB中做特征层的下采样，同时channel增大，在所述RB中对特征层充分融合后输出；以及

S313、最后，所述backbone网络输出三个特征图。