| 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')#数据集源文件存放处 | |