added training script

This commit is contained in:
2024-04-29 16:29:42 +02:00
parent 8a7dc71f0e
commit 9ac5ba921a

6
train.py Normal file
View File

@ -0,0 +1,6 @@
from ultralytics import YOLO
# Load a model
model = YOLO('yolov8.yaml')
results = model.train(data='data/data.yaml', epochs=10, imgsz=640, device='mps', workers=0)