Add conversion to FEN

2025-12-22 16:30:07 +01:00
parent 86dea774e4
commit 0aaea36586
12 changed files with 246 additions and 105 deletions
--- a/rpi/training/training_v11_n.py
+++ b/rpi/training/training_v11_n.py
@@ -1,17 +1,17 @@
 from ultralytics import YOLO

 def main():
-    model = YOLO("models/unified-nano.pt")
+    model = YOLO("models/yolo11n-seg.pt")
    model.train(
-        data="./datasets/pieces/unified/data.yaml",
+        data="./datasets/edges/data.yaml",
        epochs=150,
        patience=20,
        imgsz=640,
-        batch=18,
+        batch=12,
        save_period=10,
        project="result",
-        name="unified-nano-refined",
-        exist_ok=True,
+        name="edges-nano",
+        exist_ok=False,
        device = 0
    )