训练数据

yolo detect train data=coco128.yaml model=yolov8n.pt epochs=100 imgsz=640 device=mps

coco128.yaml


# Train/val/test sets as 1) dir: path/to/imgs, 2) file: path/to/imgs.txt, or 3) list: [path/to/imgs1, path/to/imgs2, ..]
path: /Users/carl/workspace/python/yolov5/sample/tunnel  # dataset root dir
train: train2  # train images (relative to 'path') 128 images
val: valid  # val images (relative to 'path') 128 images
test:  # test images (optional)

# Classes
names:
  0: KL
  1: SS
  2: TL
  3: XB
  4: XBHLF

目录结构

如果训练异常中断了，需要重新训练，可以使用resume参数

yolo detect train data=coco128.yaml model=yolov8n.pt epochs=100 imgsz=640 device=mps resume=True