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

布满荆棘的人生 2022-09-02 01:29 642阅读 0赞

# 目录 #

### 0.摘要 ###

### 1.开源项目github链接 ###

### 2.数据集详细情况 ###

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

### 4.测试效果 ###

### 5.说明 ###

## 0.摘要 ##

本文开源1个工作服(反光衣)检测数据集(含标注)和预训练模型，此项目已经上传github，欢迎star。

工作服(反光衣)-安全帽检测(实用的目标检测) qq群: 980489677 qq2群：710514100

CVAT使用标注教程：https://blog.csdn.net/LEILEI18A/article/details/113385510

此项目可用于施工区域or危险区域检测人员！！！

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xFSUxFSTE4QQ_size_16_color_FFFFFF_t_70][]

## 1.开源项目github链接 ##

https://github.com/gengyanlei/reflective-clothes-detect

开源模型为yolov5s模型

## 2.数据集详细情况 ##

本文开源工作服(反光衣)检测数据集图像1083张(含xml标注)：包含不同颜色、(警察-环保-工地-海边(救生衣))等场景；

CVAT使用标注教程：https://blog.csdn.net/LEILEI18A/article/details/113385510

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

(1)基于yolov4训练反光衣1083张图像，获得反光衣检测模型；

(2)基于(1)的模型对SHWD（[https://github.com/njvisionpower/Safety-Helmet-Wearing-Dataset][https_github.com_njvisionpower_Safety-Helmet-Wearing-Dataset]）数据集进行类别扩充，获得4类标注(安全帽2类，反光衣2类，整体人\[由coco模型提供\])

(3)基于SHWD数据集训练即可！

注意：

yolov4 可以对空label的图像进行训练；

yolov5 也可以对空label的图像进行训练；

## 4.测试效果 ##

最新5类检测效果：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xFSUxFSTE4QQ_size_16_color_FFFFFF_t_70][]

## 5.说明 ##

(1)本数据集仅学术探索！  
————————————————  
版权声明：本文为CSDN博主「LEILEI18A」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。  
原文链接：https://blog.csdn.net/LEILEI18A/article/details/108694753

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L0xFSUxFSTE4QQ_size_16_color_FFFFFF_t_70]: /images/20220829/ff15ec7af871420ea0c7fbc156b2a8af.png
[https_github.com_njvisionpower_Safety-Helmet-Wearing-Dataset]: https://github.com/njvisionpower/Safety-Helmet-Wearing-Dataset