安全帽反光衣检测识别数据集和yolov5模型

                                                            安全帽-工作服(反光衣)检测数据集和yolov4-5检测模型

目录

0.摘要

1.开源项目github链接

2.数据集详细情况

3.工作服(反光衣)数据集扩充方案

4.测试效果

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)本数据集仅学术探索!

猜你喜欢

转载自blog.csdn.net/LEILEI18A/article/details/108694753