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里面作者已经放好了示例命令