[发明专利]一种基于YOLOv5的焊接面罩佩戴识别方法及系统

权利要求书

1.一种基于YOLOv5的焊接面罩佩戴识别方法，其特征在于，包括：

(1)采集焊接面罩数据、人体、人脸和弧光数据，标注类别并制作数据集；

(2)对YOLOv5从轻量化模块替换和特征融合重构两方面进行轻量化改进，构建得到WMU-YOLO模型，利用数据集对WMU-YOLO模型进行训练；

(3)将待检测图片输入至WMU-YOLO，依次判断检测到的人体框与弧光框、人脸框、焊接面罩框的相交情况是否满足各自设定阈值，识别焊工在工作时是否佩戴焊接面罩。

2.根据权利要求1所述的方法，其特征在于，步骤(2)包括：

以YOLOv5s为基准模型，在YOLOv5的结构中，将Backbone和Neck中所有的Conv模块替换为Ghost-Conv模块，将部分C3模块替换为Ghost-C3模块，将Neck的特征融合方式由拼接替换为Bi-FPN中的快速归一化融合方法，并添加一个额外的跨层级特征信息融合路径，构建得到WMU-YOLO模型，利用数据集对WMU-YOLO进行训练，Ghost-Conv模块结构中，产生Ghost特征的线性操作为深度卷积；Ghost-C3模块在Bottleneck中添加深度卷积模块和Ghost-Conv模块。

3.根据权利要求1或2所述的方法，其特征在于，步骤(1)包括：

采集能够涵盖焊接场景、作业类型和目标尺度变化的焊接面罩数据和弧光数据，其中，焊接面罩数据包括头戴式和手持式，对于手持式焊接面罩，只有当焊工将整个脸部遮住才将其进行标记，弧光数据收集与焊接面罩收集同步进行，在焊枪焊条或焊丝与被焊器件的接触区域以小段多标的方式对焊接弧光进行标记；

从焊接图像和行人公共数据集CrowdHuman采集人体和人脸数据，在标记焊接面罩和弧光数据的同时，对图像中出现的焊工及人脸进行标注，焊工的焊接姿势涉及到蹲姿、坐姿、躺姿和站姿；

按照比例将图像和标注文件制作成数据集。

4.根据权利要求3所述的方法，其特征在于，步骤(3)包括：

将待检测图片输入至WMU-YOLO模型中，运行得到待检测图片中焊接面罩、弧光、人体和人脸的目标位置、目标置信度和目标类别；

以φ(X,Y)为边界框X和Y之间的相交面积与边界框Y的面积之比，设定阈值α，当人体框与弧光框之间的相交情况φ小于等于α，则认为人体框对应的焊工未处于焊接环境，此时过滤掉与该人体框相交的其他框，否则，判断人体框对应的焊工正从事焊接作业，进一步识别人体框对应的焊工是否佩戴焊接面罩；

判断人体框与人脸框之间的相交情况φ是否满足设定阈值β，若小于等于设定阈值β，则进一步判断是否佩戴焊接面罩，否则，过滤掉与该人体框相交的其他框，表示人体框对应的焊工脸部暴露在焊接环境中且未佩戴焊接面罩；

判断人体框和焊接面罩框的相交情况φ是否满足设定阈值γ，若大于设定阈值γ，则过滤掉与该人体框相交的其他框，表示人体框对应的焊工在进行焊接作业过程中正确佩戴焊接面罩，反之表示人体框对应的焊工在进行焊接作业过程中未正确佩戴焊接面罩。

5.根据权利要求4所述的方法，其特征在于，α设定为0，β、γ均设为0.5。

6.一种基于YOLOv5的焊接面罩佩戴识别系统，其特征在于，包括：

数据采集模块，用于采集焊接面罩数据、人体、人脸和弧光数据，标注类别并制作数据集；

模型构建训练模块，用于对YOLOv5从轻量化模块替换和特征融合重构两方面进行轻量化改进，构建得到WMU-YOLO模型，利用数据集对WMU-YOLO模型进行训练；

合规性验证识别模块，用于将待检测图片输入至WMU-YOLO，依次判断检测到的人体框与弧光框、人脸框、焊接面罩框的相交情况是否满足各自设定阈值，识别焊工在工作时是否佩戴焊接面罩。