安全帽-工作服(反光衣)检测数据集和yolov4-5检测模型
目录
0.摘要
本文开源1个工作服(反光衣)检测数据集(含标注)和预训练模型,此项目已经上传github,欢迎star。
工作服(反光衣)-安全帽检测(实用的目标检测) qq群: 980489677 qq2群:710514100
CVAT使用标注教程:数据标注工具CVAT教程_LEILEI18A的博客-CSDN博客_cvat标注工具
此项目可用于施工区域or危险区域检测人员!!!
1.开源项目github链接
https://github.com/gengyanlei/reflective-clothes-detect
开源模型为yolov5s模型
2.数据集详细情况
本文开源工作服(反光衣)检测数据集图像1083张(含xml标注):包含不同颜色、(警察-环保-工地-海边(救生衣))等场景;
CVAT使用标注教程:数据标注工具CVAT教程_LEILEI18A的博客-CSDN博客_cvat标注工具
3.工作服(反光衣)数据集扩充方案
(1)基于yolov4训练反光衣1083张图像,获得反光衣检测模型;
(2)基于(1)的模型对SHWD(右击新建标签页打开链接)数据集进行类别扩充,获得4类标注(安全帽2类,反光衣2类,整体人[由coco模型提供])
(3)基于SHWD数据集训练即可!
注意:
yolov4 可以对空label的图像进行训练;
yolov5 也可以对空label的图像进行训练;
4.测试效果
最新5类检测效果:
5.说明
(1)本数据集仅学术探索!