TP-Ai-Alg/ultralytics-main/TRain.md

yolo训练自己的数据

1、制作数据集

①：标注工具：labelme

②：标注完后使用脚本将xml转为yolo格式（txt）

③：将转好格式的数据划分为训练集（train）和验证集（val），比例为7：3。

④：参考coco128数据的文件格式，将自己的数据文件夹层次改为和coco128的相似，和train2017同级及一个val文件夹，存放验证集数据，coco128中val

2、修改配置文件

①：复制coco128.yaml文件到ultralytics-main目录，并改名为mydata.yaml

②：修改 path 后面的路径为你自己放数据的路径。例：path: /home/tp/datasets/mydata

③：修改 train 和 val后面的路径，如果子的数据目录结构和coco128结构完全相同，那么你可以不需要改动。

例：train: images/train2017
    val: images/val

④：将 name 后面的标签改为自己标注时的标签。或者这个0要与xml转txt时对应的标签名一致

例：0: smoke
    1: rmb

3、训练

①：进入ultralytics-main目录

②：训练参数解释：

data=coco128.yaml、model=yolov8n.pt       将参数路径改为实际文件路径

epochs=100                                训练100批次

batch=16                                  batch_size大小为16

save_period=50                            隔固定周期保存模型，小于1，为禁用,默认禁用

③：执行以下命令

python yolo detect train data=mydata.yaml model=yolov8n.pt epochs=100 imgsz=640