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

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

【基于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

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

更多推荐