同时有安全帽识别、反光衣识别、锥桶识别的YOLOv5数据集(有ui界面)
可以同时识别反光衣、安全帽、锥桶、人这几个类别的yolov5模型,识别率基本稳定90以上,还用qt designer 和pyqt设计了一个简单的ui界面
·
看了一下网上能找到的数据集基本上都是只有安全帽识别或者只有反光衣识别的,于是自己标注了一个同时有安全帽、反光衣、人、锥桶(不想要锥桶的可以删掉)的数据集。
数据集和源码链接见视频简介:
【基于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以上。
准确率和召回率:
训练过程:
最终识别效果:
识别效果视频:
总结:
欢迎探讨评论,评论看到了就会回

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐
所有评论(0)