【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文件
【YOLOV5模型训练】-模型训练需要改动的几个地方_第1张图片
第一个地方改为自己新建的数据集
第二个地方改为自己想要到训练迭代次数

python3 train.py

即可运行

你可能感兴趣的:(YOLO系列,深度学习,python)