当前位置:   article > 正文

助力安全生产,基于轻量级yolov5n模型开发构建工地穿戴安全帽、反光衣检测识别系统

yolov5n

在日常的生产作业活动中经常会出现因为工人未按照安全生产规范要求进行作业导致发生安全事故的问题,本文的核心目的就是想要结合深度学习模型开发构建智能检测模型,助力安全生产,首先看下效果图:

 可以看到:可视化系统端集成融合了很多组件,实现了全面的功能。

接下来看下数据集:

 数据主要是简单从互联网爬取人工标注处理的。

当然了网上也有很多开源的数据集也都是可以使用的,感兴趣的话可以去找一下。

训练数据配置文件如下:

  1. # Dataset
  2. path: ./dataset
  3. train:
  4. - images/train
  5. val:
  6. - images/test
  7. test:
  8. - images/test
  9. # Classes
  10. names:
  11. 0: reflective_clothes
  12. 1: hat
  13. 2: person
  14. 3: other_clothes

模型文件配置如下:

  1. # YOLOv5
    声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/AllinToyou/article/detail/147483
    推荐阅读
    相关标签