手把手教你进行安全帽的佩戴检测（附数据集+代码演示+实验结果）

一、数据集和代码的准备

二、训练过程

三、结果演示

那让我们开始吧！

一、数据集和代码的准备

数据集：链接：https://pan.baidu.com/s/1tN7g26s8DRgAKrn6FK3wjg?pwd=8858
提取码：8858
这个数据集是我花了三个小时搞定的！很辛苦的！

代码：

我们这次选择yolov5模型作为基线训练

首先去yolov5的官方项目里下载代码：https://github.com/ultralytics/yolov5

二、训练过程

1、数据集放在datasets文件夹中

datasets属于放置数据集的地方，位于C:\Users\Lenovo\PycharmProjects中（这是我的电脑位置哈，跟您的不一定一样，反正位于PycharmProjects中），属于项目的同级文件夹。

2、打开pycharm，在data文件夹里新建一个helmet.yaml，如图所示

打开helmet.yaml，按照如下图所示的进行配置：

注意：train和val需要用到本人电脑的绝对路径，根据自己数据集所放位置来。

3、在项目里找到train.py ，在--data这里放上我们刚刚写好的helmet.yaml的路径

4、开始训练，train.py点击“运行”

由结果可知，准确率97.9%，召回率91.5%，效果还是很不错的。

5、测试训练的模型

找到detect.py ,将这里的权重路径换成我们刚刚训练好的那个模型的路径

然后找到data/images文件夹，在这个文件夹中放上我们想检测的图片

放完后我们就可以执行detece.py了，执行后会打印出检测后图片的路径，我们直接打开就可以看到了。

三、结果演示

图片：

视频：

好的，就到这里了，感谢大家！

手把手教你进行安全帽的佩戴检测（附数据集+代码演示+实验结果）相关推荐

干货：手把手教你Tableau高级数据分析功能（附数据集）
原文标题:A Step-by-Step Guide to learn Advanced Tableau – for Data Science and Business Intelligence Pro ...
实战：手把手教你实现用语音智能控制电脑 | 附完整代码
作者 | 叶圣出品 | AI科技大本营(ID:rgznai100) 导语:本篇文章将基于百度API实现对电脑的语音智能控制,不需要任何硬件上的支持,仅仅依靠一台电脑即可以实现.作者经过测试,效果 ...
手把手教你Tableau高级数据分析功能（附数据集）
原文标题:A Step-by-Step Guide to learn Advanced Tableau – for Data Science and Business Intelligence Pro ...
手把手教你入门深度强化学习（附链接代码）
来源:机器之心本文约2600字,建议阅读10分钟. GitHub 开源了一份深度强化学习的教程,总结了从 DQN 到彩虹模型的理论和代码实现. 深度强化学习在机器学习领域的热度一直很高.最近,Git ...
转：干货：手把手教你Tableau高级数据分析功能（附数据集）
地址:https://blog.csdn.net/Tw6cy6uKyDea86Z/article/details/79694011 原文标题:A Step-by-Step Guide to learn ...
简单有趣的 NLP 教程：手把手教你用 PyTorch 辨别自然语言（附代码）
简单有趣的 NLP 教程:手把手教你用 PyTorch 辨别自然语言(附代码) 雷锋网(公众号:雷锋网)按:本文作者甄冉冉,原载于作者个人博客,雷锋网已获授权. 最近在学pyTorch的实际应用 ...
手把手教你开发photoshop面板插件（附demo和工具）
手把手教你开发photoshop面板插件(附demo和工具) 一.前言二.插件演示三.目录文件介绍 3.1 插件安装 3.2 开启ps开发模式 3.3 插件文件介绍 3.4 manifest 文件 ...
B站教学手把手教你使用YOLOV5之口罩检测项目最全记录详解 ( 深度学习 / 目标检测 / pytorch )
目录一.环境搭建 pytorch的下载测试(cmd窗口中) pycharm下测试(要配置pycharm中的虚拟环境) 二.数据标注下载labor image 使用labelimg进行图片标注划 ...
手把手教你申请计算机软件著作权（2）—— 生成代码文件身份证明
#今天也是咸鱼的一天~ 上一篇关于软著申请的博客手把手教你申请计算机软件著作权(1)里有说,要求提交一份代码的前三十页和后三十页.在这篇博客里,就详细叙述一下如何提交代码文件. 手把手教你申请软件著作 ...

手把手教你进行安全帽的佩戴检测（附数据集+代码演示+实验结果）

手把手教你进行安全帽的佩戴检测（附数据集+代码演示+实验结果）相关推荐

最新文章

热门文章