看了一下网上能找到的数据集基本上都是只有安全帽识别或者只有反光衣识别的,于是自己标注了一个同时有安全帽、反光衣、人、锥桶(不想要锥桶的可以删掉)的数据集。

数据集和源码链接见视频简介:

【基于yolov5的反光衣、安全帽识别】
https://www.bilibili.com/video/BV1W14y1J7pP/?share_source=copy_web&vd_source=2e7cf29355d6d43bff771e3dcbd2645d

目录

-images
-labels(标注好的yolo txt格式)

类别

数据集包含4个类别

  1. 安全帽(hat)
  2. 反光衣(reflect)
  3. 人(person)
  4. 防护锥桶(fanghu)
    可以同时识别安全帽、反光衣和人;利用YOLOv5训练后的准确率达到95以上。
准确率和召回率:

在这里插入图片描述

训练过程:在这里插入图片描述
最终识别效果:

在这里插入图片描述
在这里插入图片描述
识别效果视频:

总结:

欢迎探讨评论,评论看到了就会回

Logo

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

更多推荐