同时有安全帽识别、反光衣识别、锥桶识别的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以上。
准确率和召回率:
训练过程:
最终识别效果:
识别效果视频:
总结:
欢迎探讨评论,评论看到了就会回

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。
更多推荐
所有评论(0)