Making it installabel package and switching to segmentation mode

2025-12-05 15:51:16 +02:00
parent 9011276584
commit 310e0b2285
20 changed files with 667 additions and 56 deletions
--- a/src/utils/config_manager.py
+++ b/src/utils/config_manager.py
@@ -56,7 +56,7 @@ class ConfigManager:
                ],
            },
            "models": {
-                "default_base_model": "yolov8s.pt",
+                "default_base_model": "yolov8s-seg.pt",
                "models_directory": "data/models",
            },
            "training": {