【YOLOV5从开发到部署实战】四、模型训练

首先我们去到YOLOV5的官方仓库下：https://github.com/ultralytics/yolov5 看到一个绿色的按钮 Code，然后点击Download ZIP，就可以拿到代码了。如果你进不去这个仓库的网址，你可以查看这个方法：github网址无法访问怎么办？ 把yolov5这个仓库链接代码导入到Gitee，你就可以下载了。解压缩代码以后，进入到YOLOv5的代码文件夹：

加入数据

我们把上一步准备好的数据集文件夹dataset直接复制或者剪切到yolov5的文件夹下。

修改参数

进入到yolov5文件夹下的models下，把每个**.yaml**结尾的文件里面的第4行修改为你的类别数。

开始训练

完成以上所有工作，我们就可以在yolov5文件夹下打开cmd，输入：

# img是输入网络图片大小，batch是每批次读入图片数量，epochs是训练的轮次，data是我们准备的数据集文件夹，cfg是模型配置文件，weights是采取的预训练模型，device是使用什么设备，0就是gpu，使用cpu就写cpu
# 注意：