ultralytics/train.py

from ultralytics import YOLO

if __name__ =='__main__':
    model=YOLO("/home/lab/LJ/wampee/ultralytics/ultralytics/cfg/models/v8/yolov8.yaml")#.yaml配置文件
    #model.load("yolov8l.pt")#.pt文件

    model.train(data="/home/lab/LJ/wampee/tomatoOD_test.yaml",epochs=400,batch=8,device = '3')#数据集源文件存放处