基于yolov5的口罩检测

首先根据上个视频搭建yolov5环境

将口罩检测的数据集copy到目录

制作口罩数据的yaml文件，命名mask.yaml

修改模型的yaml文件，命名mask.yaml

训练数据，配置对应路径

开始训练，我在cpu上迭代100次用时7hours

训练完之后，有对应的文件，找到best.pt

接下来就是进行检测的时候了

检测结果

结果还是不太理想，这里我迭代了100次，建议迭代300次

到此，一名小白的口罩检测基本完成

接下来训练更换的best.pt，让自己的模型更好更快更准确！！！

谨以此文记录小白的学习！