Fix wrong import path in ultralytics mobile_lib (sony#983)

reuvenperetz · Mar 7, 2024 · 34dba6b · 34dba6b
1 parent f526862
commit 34dba6b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tutorials/quick_start/pytorch_fw/ultralytics_lib/model_lib.py b/tutorials/quick_start/pytorch_fw/ultralytics_lib/model_lib.py
@@ -31,7 +31,7 @@
 from tutorials.quick_start.common.constants import MODEL_NAME, BATCH_SIZE, COCO_DATASET, VALIDATION_DATASET_FOLDER, \
     MODULE_REPLACER
 from tutorials.quick_start.common.results import DatasetInfo
-from tutorials.mct_model_garden.yolov8.yolov8_preprocess import yolov8_preprocess_chw_transpose
+from tutorials.mct_model_garden.models_keras.yolov8.yolov8_preprocess import yolov8_preprocess_chw_transpose
 from model_compression_toolkit.core import FolderImageLoader
 from model_compression_toolkit.core.pytorch.back2framework.pytorch_model_builder import PytorchModel