from ultralytics import YOLO # 加载预训练模型 model = YOLO('../yolov8n.pt') # 开始训练 model.train( data='./yolov8.yaml', # 数据配置文件路径 epochs=320, # 训练轮数 batch=-1, # 批量大小 imgsz=640, # 输入图片大小 device=0 # 使用的设备(0 表示 GPU,'cpu' 表示 CPU) )