一种安全帽佩戴检测的图像数据集冗余清洗方法技术

本发明专利技术涉及一种安全帽佩戴检测的图像数据集冗余清洗方法，包括以下步骤：使用目标检测算法YOLO V4算法在VOC数据集上进行训练，选择最优结果的模型文件进行保存；利用FFmpeg将摄像头采集到的视频解析成图像；解析完成的图像按照顺序输入训练完成的目标检测模型，检测结果仅保留检测到人的图像；判断前后两张图检测到的人员数量是否一致；不一致，则将两张图像均保存；一致，则计算两张图像中人员各个人员位置中心点坐标；计算各个对应中心点之间的距离之和；判断计算距离之和是否大于设定阈值；判断结果为否，则删除其中一张图；判断结果为是，则保留两张图。本发明专利技术的优点：提升安全帽佩戴检测图像数据冗余清洗方法的可解释性。佩戴检测图像数据冗余清洗方法的可解释性。佩戴检测图像数据冗余清洗方法的可解释性。

【技术实现步骤摘要】
一种安全帽佩戴检测的图像数据集冗余清洗方法


[0001]本专利技术涉及目标检测
，具体是指一种安全帽佩戴检测的图像数据集冗余清洗方法。

技术介绍

[0002]安全帽作为施工作业时必须佩戴的工作帽有着诸多应用场合，如建筑工地、工厂生产线以及井下采矿等。作业人员佩戴安全帽不仅是安全生产作业的规范要求，更是人生安全的重要保障。因此，安全帽佩戴检测模型有着十分广阔的应用前景。与此同时，需要安全帽佩戴检测模型准确率达到一定高度。这就要对每个具体应用场合的监控视频数据进行有针对性的安全帽佩戴检测模型训练。
[0003]随着深度学习技术的不断发展，目标检测模型作为机器视觉识别中重要方向之一，也广泛采用了基于深度学习的检测技术。通常，深度学习技术需要大量标注完成的样本作为训练集，并利用反向传播算法对其模型参数进行调优以达到预定效果。深度学习不仅要求参与训练的样本数量要多，还需要样本之间体现出差异。大量相似的样本是不利于提升深度学习模型的泛化能力，也不利于模型部署后的实际测试应用。
[0004]在安全帽佩戴检测模型实际训练时，训练集通常是通过网络监控摄像头或录像机获取视频数据并将视频数据解析成图像数据，然后利用图像标注软件平台对该图像中没有佩戴安全帽的区域进行标注。但是，当视频解析成图像是按照一定帧率将视频中的帧以图像格式保存。如果设定的帧率偏低，则原本拍摄到没有佩戴安全帽的图像可能会丢失，导致数据集收集周期变长；如果设定的帧率偏高，则可能会存在大量重复数据。为了尽快收集数据，帧率往往设置偏高。因此，需要对视频转图像后存在的冗余进行自动清除。
[0005]针对图像数据集中冗余清除，目前常用的方法有以下几种：
[0006](1)人工检测
[0007]人工检测作为最原始的方法，虽然判断图像数据集中的冗余数据较为准确，但是人工成本高。此外，人工检测主观性较强，不同标注人员之间存在差异。这些因素使得通过人工检测清洗图像数据集中冗余的方式难以及时响应项目需求。
[0008](2)结构相似性
[0009]结构相似性作为评价两张图像之间相似程度的指标常用于图像去相似任务中。但是，将其应用在安全帽佩戴监测项目中会存在因非安全帽区域变化而导致的结构相似性值变高，误将其判定为两张图存在差异，这点往往是我们不愿意看到的。
[0010]上述方法不能在安全帽佩戴检测项目中很好地实现图像数据集冗余清洗。

技术实现思路

[0011]本专利技术要解决上述技术问题，提供一种可以适用于安全帽佩戴检测的图像数据集冗余清洗方法。
[0012]为解决上述技术问题，本专利技术提供的技术方案为：一种安全帽佩戴检测的图像数
据集冗余清洗方法，包括以下步骤：
[0013]S1，使用目标检测算法YOLO V4算法在VOC数据集上进行训练，以反向传播算法进行参数调优，选择最优结果的模型文件进行保存；
[0014]S2，利用FFmpeg将摄像头采集到的视频解析成图像；
[0015]S3，解析完成的图像按照顺序输入步骤S1中训练完成的目标检测模型，检测结果仅保留检测到人的图像；
[0016]S4，判断前后两张图检测到的人员数量是否一致；
[0017]S5，如果步骤S4中判断结果为不一致，则将两张图像均保存；
[0018]S6，如果步骤S4中判断结果为一致，则计算两张图像中人员各个人员位置中心点坐标，将两张图之间中心点坐标距离最近的两个中心点进行对应；
[0019]S7，计算各个对应中心点之间的距离之和；
[0020]S8，判断计算距离之和是否大于设定阈值；
[0021]S9，如果S8中判断结果为否，则删除其中一张图；
[0022]S10，如果S8中判断结果为是，则两张图像均保存。
[0023]本专利技术具有如下优点：
[0024]1、减少因现场设备等与安全帽无直接关联的物体运动变化而造成安全帽佩戴检测图像数据集非冗余误判；
[0025]2、本专利技术使用的目标检测模型无需使用大量安全帽检测现场实际数据进行模型训练；
[0026]3、对作业人员出现的无效图像进行过滤；
[0027]4、提升安全帽佩戴检测图像数据冗余清洗方法的可解释性。
[0028]作为改进，所述步骤S3中输出中仅保留检测到人的结果数据替换为VOC数据集中标注为person的结果数据。
[0029]作为改进，所述步骤S2具体为利用FFmpeg将摄像头拍摄到的视频文件按照设定的帧率解析成图像格式保存。
[0030]作为改进，所述步骤S5具体为如果两张图像检测到的人员数量信息之间不一致，则判定两张图不相似，不作为冗余对其删除。
[0031]作为改进，所述步骤S6具体为如果两张图像检测到的人员数量信息之间一致，则分别计算检测结果中各个人员位置中心点坐标，前一张图像中对应的各个人员中心点坐标分别与后一张图像中距离最近的人员中心坐标点进行对应。
[0032]作为改进，所述步骤S9具体为如果中心坐标点间距离之和小于等于阈值则表示两张图之间相似，其中一张图是冗余并将其删除。
[0033]作为改进，所述步骤S10具体为如果中心坐标点间距离之和大于阈值则表示两张图之间不相似，不存在冗余，两张图像均保存。
附图说明
[0034]图1是本专利技术一种安全帽佩戴检测的图像数据集冗余清洗方法的步骤流程示意图。
具体实施方式
[0035]下面结合附图对本专利技术做进一步的详细说明。
[0036]结合附图1，
[0037]实施列一
[0038]一种安全帽佩戴检测的图像数据集冗余清洗方法，包括以下步骤：
[0039]S1，使用目标检测算法YOLO V4算法在VOC数据集上进行训练，以反向传播算法进行参数调优，选择最优结果的模型文件进行保存；
[0040]S2，利用FFmpeg将摄像头采集到的视频解析成图像；
[0041]S3，解析完成的图像按照顺序输入步骤S1中训练完成的目标检测模型，检测结果仅保留检测到人的图像；
[0042]S4，判断前后两张图检测到的人员数量是否一致；
[0043]S5，如果步骤S4中判断结果为不一致，则将两张图像均保存；
[0044]S6，如果步骤S4中判断结果为一致，则计算两张图像中人员各个人员位置中心点坐标，将两张图之间中心点坐标距离最近的两个中心点进行对应；
[0045]S7，计算各个对应中心点之间的距离之和；
[0046]S8，判断计算距离之和是否大于设定阈值；
[0047]S9，如果S8中判断结果为否，则删除其中一张图；
[0048]S10，如果S8中判断结果为是，则两张图像均保存。
[0049]实施列二
[0050]一种安全帽佩戴检测的图像数据集冗余清洗方法，包括以下步骤：
[0051]S1，使用目标检测算法YOLO V4算法在VOC数据集上进行训练，以反向传播本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种安全帽佩戴检测的图像数据集冗余清洗方法，其特征在于：包括以下步骤：S1，使用目标检测算法YOLO V4算法在VOC数据集上进行训练，以反向传播算法进行参数调优，选择最优结果的模型文件进行保存；S2，利用FFmpeg将摄像头采集到的视频解析成图像；S3，解析完成的图像按照顺序输入步骤S1中训练完成的目标检测模型，检测结果仅保留检测到人的图像；S4，判断前后两张图检测到的人员数量是否一致；S5，如果步骤S4中判断结果为不一致，则将两张图像均保存；S6，如果步骤S4中判断结果为一致，则计算两张图像中人员各个人员位置中心点坐标，将两张图之间中心点坐标距离最近的两个中心点进行对应；S7，计算各个对应中心点之间的距离之和；S8，判断计算距离之和是否大于设定阈值；S9，如果S8中判断结果为否，则删除其中一张图；S10，如果S8中判断结果为是，则两张图像均保存。2.根据权利要求1所述的一种安全帽佩戴检测的图像数据集冗余清洗方法，其特征在于：所述步骤S3中输出中仅保留检测到人的结果数据替换为VOC数据集中标注为person的结果数据。3.根据权利要求1所述的一种安全帽...

【专利技术属性】
技术研发人员：陈轶，张文，王茂森，牛少彰，崔浩亮，冯亚辉，王让定，
申请(专利权)人：浙江砖助智连科技有限责任公司，
类型：发明
国别省市：