Testing detect

2025-12-19 10:44:11 +02:00
parent 6ae7481e25
commit 8f8132ce61
2 changed files with 9 additions and 3 deletions
--- a/src/model/yolo_wrapper.py
+++ b/src/model/yolo_wrapper.py
@@ -196,7 +196,7 @@ class YOLOWrapper:
                f"Running inference on {source} -> prepared_source {prepared_source}"
            )
            results = self.model.predict(
-                source=prepared_source,
+                source=source,
                conf=conf,
                iou=iou,
                save=save,