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

即可运行