基于全新数据集和改进YOLOv5的防护设备佩戴检测方法
摘要:
本发明公开了一种基于全新数据集和改进YOLOv5的防护设备佩戴检测方法。该方法包括:构建全新PPE检测数据集;将收集到的初始视频数据集和图片数据集作为不完全数据集A,对不完全数据集A进行人工手动标注;构建成改进损失函数模型YOLOv5‑SEIoU;基于改进损失函数模型YOLOv5‑SEIoU训练标注后的不完全数据集A,获得模型A;根据模型A对于不同标注类别的检测精度,完成各标注类别的标注集的正确标注;将正确标注的各类别标注集合并至标注后的不完全数据集A作为不完全数据集B,将各标注类别的额外数据集中的待标注集输入模型B自动标注,输出的数据集与不完全数据集B作为均衡的完全数据集。本发明提升目标检测的检测精度和速度,能够实时检测建筑工人是否穿戴PPE。
0/0