[发明专利]一种基于YOLOv5的安全护具佩戴检测方法及系统

权利要求书

1.一种基于YOLOv5的安全护具佩戴检测方法，其特征在于，包括：

获取待检测的图像数据；

将所述待检测的图像数据输入YOLOv5网络进行预测，得到特征图；所述特征图包括目标物体类别、目标物体个数和检测框位置；

根据所述目标物体类别、所述目标物体个数和所述检测框位置确定施工人员是否正确佩戴安全护具。

2.根据权利要求1所述的基于YOLOv5的安全护具佩戴检测方法，其特征在于，所述获取待检测的图像数据之前，还包括：

获取待检测的视频数据；

对所述待检测的视频数据进行分解，得到待检测的图像数据。

3.根据权利要求1所述的基于YOLOv5的安全护具佩戴检测方法，其特征在于，所述将所述待检测的图像数据输入YOLOv5网络进行预测，得到特征图，具体包括：

利用Mosaic数据增强算法对所述待检测的图像数据进行预处理，得到预处理后的图像数据；

将所述预处理后的图像数据输入所述YOLOv5网络中的特征提取网络进行特征提取，得到显著特征图；

将所述显著特征图输入所述YOLOv5网络中的特征处理网络进行特征处理，得到特征图。

4.根据权利要求3所述的基于YOLOv5的安全护具佩戴检测方法，其特征在于，所述将所述预处理后的图像数据输入所述YOLOv5网络中的所述特征提取网络进行特征提取，得到显著特征图，具体包括：

利用所述特征提取网络的Focus模块对所述预处理后的图像数据进行切片、拼接和卷积处理，得到第一特征提取图；

利用所述特征提取网络的Conv模块组对所述第一特征提取图进行卷积处理，得到第二特征提取图；

利用所述特征提取网络的BCSP模块组对所述第二特征提取图进行Bottleneck卷积和拼接处理，得到第三特征提取图；

利用所述特征提取网络的SPP模块对所述第三特征提取图进行池化和拼接处理，得到显著特征图。

5.根据权利要求3所述的基于YOLOv5的安全护具佩戴检测方法，其特征在于，所述将所述显著特征图输入所述YOLOv5网络中的特征处理网络进行特征处理，得到特征图，具体包括：

利用所述特征处理网络的Conv模块组对所述显著特征图进行卷积处理，得到第一特征处理图；

利用所述特征处理网络的Upsample模块组对所述第一特征处理图进行采样和扩充处理，得到第二特征处理图；

利用所述特征处理网络的Concat模块组对所述第二特征处理图在通道维度上进行拼接处理，得到第三特征处理图；

利用所述特征处理网络的BCSP模块组对所述第三特征处理图进行Bottleneck卷积和拼接处理，得到第四特征处理图；

利用所述特征处理网络的Detect模块对所述第四特征处理图进行卷积处理，得到特征图。

6.根据权利要求1所述的基于YOLOv5的安全护具佩戴检测方法，其特征在于，所述根据所述目标物体类别、所述目标物体个数和所述检测框位置确定施工人员是否正确佩戴安全护具，具体包括：

根据所述目标物体类别和所述检测框位置确定安全护具检测类别；所述安全护具检测类别包括安全帽检测、反光衣检测和安全绳检测；

根据所述安全护具检测类别和所述目标物体个数确定施工人员是否正确佩戴安全护具。

7.根据权利要求6所述的基于YOLOv5的安全护具佩戴检测方法，其特征在于，所述根据所述安全护具检测类别和所述目标物体个数确定施工人员是否正确佩戴安全护具，具体包括：

当所述安全护具检测类别为安全帽检测时，根据所述目标物体类别、所述目标物体个数和所述检测框位置确定安全帽佩戴结果；

当所述安全护具检测类别为反光衣检测时，根据所述目标物体类别、所述目标物体个数和所述检测框位置确定反光衣穿戴结果；

当所述安全护具检测类别为安全绳检测时，根据所述目标物体类别、所述目标物体个数和所述检测框位置确定安全绳穿戴结果；

根据所述安全帽佩戴结果、所述反光衣穿戴结果和所述安全绳穿戴结果确定施工人员是否正确佩戴安全护具。