fix: 🛠 load_video,SamPredictor,sam_model_registry import fixes for on…

…nx export script Signed-off-by: Onuralp SEZER <thunderbirdtr@gmail.com>
kadirnar · Jun 26, 2023 · 034c660 · 034c660
1 parent 1ef26e1
commit 034c660
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/metaseg/sahi_predict.py b/metaseg/sahi_predict.py
@@ -4,7 +4,7 @@
 import torch
 from PIL import Image
 
-from metaseg import SamPredictor, sam_model_registry
+from metaseg.generator import SamPredictor, sam_model_registry
 from metaseg.utils import (
     download_model,
     load_image,

diff --git a/metaseg/utils/__init__.py b/metaseg/utils/__init__.py
@@ -8,6 +8,7 @@
 from .data_utils import load_image as load_image
 from .data_utils import load_mask as load_mask
 from .data_utils import load_server_image as load_server_image
+from .data_utils import load_video as load_video
 from .data_utils import multi_boxes as multi_boxes
 from .data_utils import plt_load_box as plt_load_box
 from .data_utils import plt_load_mask as plt_load_mask