一种基于深度学习的检测工人是否佩戴安全帽的方法.pdf

(19)国家知识产权局(12)发明专利申请(10)申请公布号CN114627425A(43)申请公布日2022.06.14(21)申请号202210363867.8G06V40/10(2022.01)(22)申请日2022.04.07G06V10/764(2022.01)G06V10/82(2022.01)(66)本国优先权数据G06K9/62(2022.01)202110653367.32021.06.11CNG06N3/04(2006.01)(71)申请人珠海路讯科技有限公司G06N3/08(2006.01)地址519000广东省珠海市横琴新区宝华路6号105室-1297(72)发明人曹振华陈伟烽梁伟鑫黄中周友元钟振鸿(74)专利代理机构珠海飞拓知识产权代理事务所(普通合伙)44650专利代理师陈李青(51)Int.Cl.G06V20/40(2022.01)G06V20/52(2022.01)权利要求书3页说明书8页附图2页(54)发明名称一种基于深度学习的检测工人是否佩戴安全帽的方法(57)摘要本发明属于深度学习的计算机视觉领域，是一种基于深度学习的工人安全帽佩戴检测方法，该方法包括以下步骤：S1、获取足够数量的含有佩戴安全帽人物样本和未佩戴安全帽人物样本的安全帽检测数据集，并随即按一定比例划分为训练集和测试集；S2、使用Pytorch深度学习框架，搭建CSP‑R‑FCN双阶段目标检测模型；S3、搭建模型训练和测试系统，对CSP‑R‑FCN双阶段目标检测模型进行训练和测试，直到其获得检测安全帽佩戴的能力；S4、提取实际应用场景：工地监控的视频，逐帧进行检测并输出检测结果，完成对安全帽佩戴的检测。本发明对视频流数据进行逐帧检测，标记出图像内所有人物的头部以及他们的安全帽佩戴情况。该方法不需要预设候选框、且具有较强的鲁棒性，在各种应用场景下都有高的检测精度。CN114627425ACN114627425A权利要求书1/3页1.一种基于深度学习的检测工人是否佩戴安全帽的方法，其特征在于：所述方法包括以下步骤：S1、获取足够数量的含有佩戴安全帽人物样本和未佩戴安全帽人物样本的安全帽检测数据集，并随即按一定比例划分为训练集和测试集；所述获取安全帽检测数据集的方法，包括：1)对现有的安全帽数据集进行数据清洗，并将标签数据转换为VOC数据集的标注格式；2)提取现有的行人检测数据集，使用labelimg标注工具对所述行人检测数据进行重新标注，得到可用的安全帽检测数据集；所标注的标签数据为VOC数据集的标注格式；3)逐帧提取工地监控视频中的图片，对所提取的图片进行筛选并使用labelimg标注工具标注，得到可用的安全帽检测数据集；所标注的标签数据为VOC数据集的标注格式；所述获取的安全帽检测数据集的标签数据，所标注的位置为图像内人物的头部；所标注的类别分为五类，分别是：佩戴黄色安全帽(yellow)、佩戴红色安全帽(red)、佩戴蓝色安全帽(blue)、佩戴白色安全帽(white)和未佩戴安全帽(none)；其中，未佩戴安全帽指的就是没有佩戴安全帽，无论是光头还是受到照明的影响还是带了不是安全帽的帽子都包含在内，神经网络可以在训练过程中自行分辨哪些是佩戴安全帽的，哪些没有佩戴；S2、使用Pytorch深度学习框架，搭建CSP‑R‑FCN双阶段目标检测模型，所述CSP‑R‑FCN双阶段目标检测模型包括骨干网络、csp目标区域推荐模块、R‑FCN特征提取模块和R‑FCN分类预测模块；所述骨干网络为在ImageNet数据集上预训练好的VGG16的前四个卷积块，共13层：其中第一层、第二层、第四层、第五层、第七层、第八层、第九层、第十一层、第十二层和第十三层均为卷积层，且卷积核大小均为3*3；第三层、第六层和第十层均为步长为2的最大池化下采样层；所述csp目标区域推荐模块接在骨干网络之后，可分为三个阶段，依次分别是：残差特征提取阶段、多层特征融合阶段和目标预测阶段；所述R‑FCN特征提取模块的输入为骨干网络的最后一层，经过一个卷积核大小为3*3的卷积层后，分为置信度预测分支和位置预测分支；置信度预测分支和位置预测分支分别经过均值池化层，输出尺寸均为7*7的置信度预测特征图与位置预测特征图；所述R‑FCN分类预测模块，包括：将输入的置信度预测特征图和位置预测特征图通过位置敏感ROI池化(Position‑sensitiveROIPooling)操作，分别得到位置敏感的类别得分图和回归得分图S3、搭建模型训练和测试系统，对CSP‑R‑FCN双阶段目标检测模型进行训练和测试，直到其获得检测安全帽佩戴的能力；S4、提取实际应用场景：工地监控的视频，逐帧进行检测并输出检测结果，完成对安全帽佩戴的检测。2.根据权利要求1所述的一种基于深度学习的检测工人是否佩