from ultralytics import YOLO

# 加载预训练模型
yolo = YOLO('models/yolov8n.pt')
# 训练模型   
# data:数据集路径   
# epochs:训练轮数  
# imgsz: 图片大小   
# batch: 批次大小
# device: 使用设备 0:GPU 'cpu':CPU

yolo.train(data='train.yaml', epochs=200, imgsz=640,batch=32,device='cpu')
print("训练完成")