【YOLOV5模型训练】-模型训练需要改动的几个地方
需要修改的几个地方
- 1.编写.yaml文件
- yaml待修改的参数特别重要一定要核对准确，直接影响模型训练成功与否
- 2.修改代码部分
1.编写.yaml文件
下载完源码后需要重新编写一个.yaml文件，用了训练自己的模型 新建一个.yaml文件具体内容如下：
# Train command: python train.py --data my_dataset.yaml
train: ../my_dataset/images/train/ #训练集images
val: ../my_dataset/images/val/ # 验证集 images
# 标签个数
nc: 3
# class names
names: [ 'l1', 'l2', 'l3' ] #类别的命名yaml待修改的参数特别重要一定要核对准确，直接影响模型训练成功与否
2.修改代码部分
打开源码里面的train.py文件
在这里插入图片描述
第一个地方改为自己新建的数据集 第二个地方改为自己想要到训练迭代次数
python3 train.py即可运行
本文参与 腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。
原始发表：2024-01-17，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读
目录