5 lines
155 B
Python
5 lines
155 B
Python
from ultralytics import YOLO
|
|
model = YOLO("yolov8n.pt")
|
|
model.train(data="data.yaml", epochs=200, batch=8, device='cpu')
|
|
model.val()
|
|
print('训练完成') |