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

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

加入数据

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

修改参数

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

开始训练

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

# img是输入网络图片大小，b