From 9ac5ba921a34ff68776e48404007167f714c3837 Mon Sep 17 00:00:00 2001
From: Samuel Dobron <xdobro23@stud.fit.vut.cz>
Date: Mon, 29 Apr 2024 16:29:42 +0200
Subject: [PATCH] added training script

---
 train.py | 6 ++++++
 1 file changed, 6 insertions(+)
 create mode 100644 train.py

diff --git a/train.py b/train.py
new file mode 100644
index 0000000..bd4d2e0
--- /dev/null
+++ 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)