yolov5检测是否佩戴口罩

yolov5检测是否佩戴口罩

数据集准备

连接地址：口罩数据集

下载yolov5模型权重

yolov5 github地址
为什么要用预训练权重呢，一般是为了缩短网络的训练时间，达到更好的精度。而yolov5的5.0版本给我们提供了几个预训练权重，我们可以对应我们不同的需求选择不同的版本的预训练权重。通过如下的图可以获得权重的名字和大小信息，可以预料的到，预训练权重越大，训练出来的精度就会相对来说越高，但是其检测的速度就会越慢。预训练权重可以通过github进行下载,这里，点进去后往下拉就能找到。本次训练自己的口罩数据集用的预训练权重为yolov5s.pt

模型训练

• 配置数据集文件
  复制文件VOC.yarml，新建mark.yaml
• 配置训练模型
  复制yolov5s.yaml新建文件yolov5s-mask.yaml