[发明专利]基于改进YOLOv3算法的非机动车头盔佩戴检测方法

权利要求书

1.一种基于改进YOLOv3算法的非机动车头盔佩戴检测方法，其特征在于，包括以下步骤：

1)以交通监控设备提供的视频流为基础，通过数据增强的方法建立训练数据集；

2)构建改进的YOLOv3目标检测算法模型；

3)将训练数据集送入改进的YOLOv3算法模型进行训练，直至模型收敛，即模型的损失函数值低于预先设定的阈值；

4)将实时交通监控视频流输入训练好的改进的YOLOv3算法模型中，检测视频流中非机动车、骑乘人员和非机动车安全头盔的位置，以确定非机动车骑乘人员是否佩戴头盔，若有非机动车骑乘人员未佩戴头盔，则进行标记；对于同一辆非机动车上检测到两个以上骑乘人员时，会进行超载标记。

2.根据权利要求1所述的一种基于改进YOLOv3算法的非机动车头盔佩戴检测方法，其特征在于：所述步骤1)使用交通监控设备录制的道路交通视频作为样本数据，所述道路交通视频的分辨率为1920x1080；将样本数据以每秒25帧转换为图像序列，并每10帧截取一张视频图像作为图像数据集，剔除其中不包含非机动车的图像，对得到的图像通过数据增强的方法进行数据扩充，所述数据增强方法具体包括图像旋转、目标遮挡和增加模拟噪声，图像旋转是指将原始图像以顺时针90°、180°和270°进行旋转得到新的图像，目标遮挡是指对原始图像中目标物体的不同部分设置黑色矩形块进行遮挡从而得到新的图像，增加模拟噪声是指使用现有的雨雾噪声算法给原始图像加上雨雾模拟噪声，对于经过数据增强得到的训练数据集，使用LabelImg软件进行多标签标注，自动生成相应的xml格式标注文件，其中包含对象名和边界框的坐标信息，类别为非机动车、骑乘人员和非机动车头盔。

3.根据权利要求1所述的一种基于改进YOLOv3算法的非机动车头盔佩戴检测方法，其特征在于：所述步骤2)构建改进的YOLOv3模型包括特征提取网络模块、空间金字塔模块、特征融合模块及多分类器模块；所述特征提取网络模块采用Darknet53网络，输入尺寸为256x256的三通道原始图像，Darknet53网络输出三种尺寸分别为32x32、16x16、8x8的特征图；所述空间金字塔模块对特征提取网络模块输出的尺寸为8x8的特征图进行最大池化操作并进行拼接得到池化特征图；所述特征融合模块将池化特征图和特征提取网络模块得到的特征图进行concat操作完成融合；所述多分类器模块采用logistic函数对特征融合模块得到的特征图进行分类检测，得到最终的目标检测结果。

4.根据权利要求1所述的一种基于改进YOLOv3算法的非机动车头盔佩戴检测方法，其特征在于：所述步骤2)改进的YOLOv3目标检测算法模型的损失函数J(θ，X，Y)用损失函数loss代替，表达式如下公式所示：

公式中λ_coord表示坐标损失的权重系数，λ_nobj表示网格预测类别的权重系数，s²表示图像划分的网格数量，B表示每个网格中包含的预测框数量，当第i个网格的第j个预测框为正确预测框时，为1，为0，反之为0，为1，x_i、y_i分别表示由第i个网络负责的目标物体的真实标注的中心点坐标，表示由第i个网络负责的目标物体的预测框的中心点坐标，ω_i、h_i表示由第i个网络负责的目标物体的真实标注的宽度、长度，表示由第i个网络负责的目标物体的预测框的宽度、长度，C_i表示第i个网络负责的目标物体的真实分类结果，表示第i个网络负责的目标物体的预测分类结果，p_i(c)表示第i个网络负责的目标物体属于分类类别c的真实概率，表示第i个网络负责的目标物体属于分类类别c的预测概率，class表示目标物体的类别标签集合。

5.根据权利要求1所述的一种基于改进YOLOv3算法的非机动车头盔佩戴检测方法，其特征在于：所述步骤3)将训练数据集中的xml标注文件进行解析得到train.txt和val.txt文件，使用训练数据集对改进的YOLOv3算法模型进行迭代训练，当模型的损失函数loss的值小于预先设定的阈值时，停止训练，保存模型参数。

6.根据权利要求1所述的一种基于改进YOLOv3算法的非机动车头盔佩戴检测方法，其特征在于：所述步骤4)包括以下步骤：

4.1)将实时交通监控视频流，并以每秒25帧转换为视频帧图像序列，并每10帧截取一张视频图像送入改进的YOLOv3算法模型；

4.2)改进的YOLOv3算法模型对图像中的非机动车、骑乘人员和头盔区域进行检测；

4.3)对于同一非机动车区域，若检测到两个以上的骑乘人员区域与该非机动车区域重叠，则说明该非机动车超载，并根据检测得到的区域坐标将超载非机动车和骑乘人员区域裁剪下来，进行超载标记；

4.4)对于未超载的非机动车，统计其图像区域内检测到的头盔区域数量，若头盔区域数量与骑乘人员区域数量不一致，说明该非机动车有骑乘人员未佩戴安全头盔，进行标记。