File tree Expand file tree Collapse file tree 3 files changed +2
-18
lines changed
Expand file tree Collapse file tree 3 files changed +2
-18
lines changed Original file line number Diff line number Diff line change @@ -24,15 +24,13 @@ jobs:
2424 python -m pip install --upgrade build
2525 - name : Build sdist
2626 run : |
27- sed -i '/omz_tools/d' src/python/pyproject.toml
2827 python -m build --sdist src/python/
2928 - uses : actions/upload-artifact@v4
3029 with :
3130 name : artifact-sdist
3231 path : src/python/dist/*.tar.gz
3332 - name : Build wheel
3433 run : |
35- sed -i '/omz_tools/d' src/python/pyproject.toml
3634 python -m build --wheel src/python/
3735 - uses : actions/upload-artifact@v4
3836 with :
Original file line number Diff line number Diff line change @@ -171,21 +171,8 @@ def __init__(
171171 )
172172 self .model = core .read_model (self .model_path , weights_path )
173173 return
174- if isinstance (model , str ):
175- from omz_tools .models import OMZModel , list_models
176-
177- if model in list_models ():
178- omz_model = OMZModel .download (
179- model ,
180- precision = precision ,
181- download_dir = download_dir ,
182- cache_dir = cache_dir ,
183- )
184- self .model_path = omz_model .model_path
185- log .info (f"Reading model { self .model_path } " )
186- self .model = core .read_model (self .model_path )
187- return
188- msg = "Model must be bytes, a file or existing OMZ model name"
174+
175+ msg = "Model must be bytes or a file"
189176 raise RuntimeError (msg )
190177
191178 def load_model (self ) -> None :
Original file line number Diff line number Diff line change @@ -28,7 +28,6 @@ dependencies = [
2828 " opencv-python-headless" ,
2929 " scipy>=1.5.4" ,
3030 " openvino>=2025.0" ,
31- " omz_tools @ git+https://github.com/openvinotoolkit/open_model_zoo.git@master#egg=omz_tools&subdirectory=tools/model_tools" ,
3231 " pillow" ,
3332]
3433
You can’t perform that action at this time.
0 commit comments