本文主要是介绍yolov5口罩检测实战,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
学习资料提要:手把手教你使用YOLOV5训练自己的目标检测模型-口罩检测-视频教程_搭建yolo目标检测的环境. 使用yolo-air模块来做实验-CSDN博客
在B站上有这个UP主的实操视频
一 环境安装
1.先在anaconda prompt 里面
(1)conda activate
会转为(base)环境
(2)conda create -n mask python==3.8
创建新的虚拟环境
(3)activate mask
激活虚拟环境
2.下载作者的 这个文档
我学到了一点:可以直接在这个目录导航输入cmd,进入的终端直接cd到这个文件路径下
如图:在这个终端下激活mask虚拟环境
3. 接下来,在这个虚拟环境里面安装pytorch,在官网里面找到cpu的安装pip
4.然后安装pycocotools
pip install pycocotools-windows
5.其他包的安装
pip install -r requirements.txt
pip install pyqt5
pip install labelme
6.测试;
python detect.py --source data/images/bus.jpg --weights pretrained/yolov5s.pt
在runs目录下可以找到检测之后的结果,环境安装完成!
二 数据标注
三 模型训练
1.下载作者的数据集,和项目都报错在同一个文件夹yolo_mask里面
2.修改配置:
(1)修改data里面是数据集配置文件
(2)修改model里面的模型配置文件
(3)在train.py里面作者已经放好了示例命令
这篇关于yolov5口罩检测实战的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!