added training script

1 year ago · 9ac5ba921a
parent 8a7dc71f0e
commit 9ac5ba921a
1 changed files with 6 additions and 0 deletions
--- a/train.py
+++ b/train.py
@ -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)