import yolo v8 y1

hotosm · Oct 28, 2024 · 4b79d48 · 4b79d48
1 parent 2d767b3
commit 4b79d48
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/hot_fair_utilities/preprocessing/__init__.py b/hot_fair_utilities/preprocessing/__init__.py
@@ -1 +1,2 @@
+from . import yolo_v8_v1
 from .preprocess import preprocess
diff --git a/test_yolo.py b/test_yolo.py
@@ -11,7 +11,7 @@
 
 # Reader imports
 from hot_fair_utilities import polygonize, predict, preprocess
-from hot_fair_utilities.preprocessing.yolov8_v1.yolo_format import yolo_format
+from hot_fair_utilities.preprocessing.yolov8_v1 import yolo_format
 from hot_fair_utilities.training.yolo_v8_v1.train import train as train_yolo
 
 warnings.simplefilter(action="ignore", category=FutureWarning)