发明公开
- 专利标题: 基于全新数据集和改进YOLOv5的防护设备佩戴检测方法
-
申请号: CN202310762540.2申请日: 2023-06-27
-
公开(公告)号: CN117058710A公开(公告)日: 2023-11-14
- 发明人: 王军武 , 刘一鹏 , 唐山荣
- 申请人: 武汉理工大学三亚科教创新园
- 申请人地址: 海南省三亚市崖州区崖州湾科技城用友产业园9号楼
- 专利权人: 武汉理工大学三亚科教创新园
- 当前专利权人: 武汉理工大学三亚科教创新园
- 当前专利权人地址: 海南省三亚市崖州区崖州湾科技城用友产业园9号楼
- 代理机构: 武汉开元知识产权代理有限公司
- 代理商 刘琳; 王月玲
- 主分类号: G06V40/10
- IPC分类号: G06V40/10 ; G06Q10/063 ; G06Q50/08 ; G06V10/25 ; G06V10/82 ; G06V20/40 ; G06V20/70
摘要:
本发明公开了一种基于全新数据集和改进YOLOv5的防护设备佩戴检测方法。该方法包括:构建全新PPE检测数据集;将收集到的初始视频数据集和图片数据集作为不完全数据集A,对不完全数据集A进行人工手动标注;构建成改进损失函数模型YOLOv5‑SEIoU;基于改进损失函数模型YOLOv5‑SEIoU训练标注后的不完全数据集A,获得模型A;根据模型A对于不同标注类别的检测精度,完成各标注类别的标注集的正确标注;将正确标注的各类别标注集合并至标注后的不完全数据集A作为不完全数据集B,将各标注类别的额外数据集中的待标注集输入模型B自动标注,输出的数据集与不完全数据集B作为均衡的完全数据集。本发明提升目标检测的检测精度和速度,能够实时检测建筑工人是否穿戴PPE。