概述
首先我们去到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
# 注意：最后
以上就是老迟到外套为你收集整理的【YOLOV5从开发到部署实战】四、模型训练的全部内容，希望文章能够帮你解决【YOLOV5从开发到部署实战】四、模型训练所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复