【YOLOV5模型训练】-模型训练需要改动的几个地方

云帆沧海

发布于 2024-01-17 17:16:46

6000

文章被收录于专栏：编程杂记编程杂记

需要修改的几个地方

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 删除

源码

【YOLOV5模型训练】-模型训练需要改动的几个地方

【YOLOV5模型训练】-模型训练需要改动的几个地方

需要修改的几个地方

1.编写.yaml文件

yaml待修改的参数特别重要一定要核对准确，直接影响模型训练成功与否

2.修改代码部分

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐