需要修改的几个地方
- 1.编写.yaml文件
- yaml待修改的参数特别重要一定要核对准确，直接影响模型训练成功与否
- 2.修改代码部分
1.编写.yaml文件
下载完源码后需要重新编写一个.yaml文件，用了训练自己的模型 新建一个.yaml文件具体内容如下：
代码语言：javascript复制# 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文件
第一个地方改为自己新建的数据集 第二个地方改为自己想要到训练迭代次数
代码语言：javascript复制python3 train.py即可运行
