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