看了一下网上能找到的数据集基本上都是只有安全帽识别或者只有反光衣识别的,于是自己标注了一个同时有安全帽、反光衣、人、锥桶(不想要锥桶的可以删掉)的数据集。
数据集和源码链接见视频简介:
【基于yolov5的反光衣、安全帽识别】
https://www.bilibili.com/video/BV1W14y1J7pP/?share_source=copy_web&vd_source=2e7cf29355d6d43bff771e3dcbd2645d
目录
-images
-labels(标注好的yolo txt格式)
类别
数据集包含4个类别
- 安全帽(hat)
- 反光衣(reflect)
- 人(person)
- 防护锥桶(fanghu)
可以同时识别安全帽、反光衣和人;利用YOLOv5训练后的准确率达到95以上。
准确率和召回率:
训练过程:
最终识别效果:
识别效果视频:
总结:
欢迎探讨评论,评论看到了就会回