le-yolo/src/train.py
2025-06-06 14:08:13 +08:00

5 lines
214 B
Python

from ultralytics import YOLO
model = YOLO(r"C:\workspace\le-yolo\runs\detect\train36\weights\last.pt")
model.train(data="data.yaml", epochs=100, batch=8, device='cpu', imgsz=640)
model.val()
print('训练完成')