yolov5检测是否佩戴口罩

yolov5检测是否佩戴口罩

数据集准备

连接地址:口罩数据集

下载yolov5模型权重

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

模型训练

  • 配置数据集文件
    复制文件VOC.yarml,新建mark.yamlyolov5检测是否佩戴口罩_第2张图片
  • 配置训练模型
    复制yolov5s.yaml新建文件yolov5s-mask.yaml

你可能感兴趣的:(机器学习,深度学习,人工智能,神经网络)