OpenVINO#
Module for otx.core.ov.
OMZ wrapper-related code for otx.core.ov.
- class otx.core.ov.omz_wrapper.NameSpace(**kwargs)#
NameSpace class for otx.core.ov.omz_wrapper.
- otx.core.ov.omz_wrapper.convert_model(model, download_dir='/home/runner/.cache/otx/omz', output_dir='/home/runner/.cache/otx/omz', precisions=None, force=False, *args)#
Converting model for OMZ wrapping.
- otx.core.ov.omz_wrapper.download_model(model, download_dir='/home/runner/.cache/otx/omz', precisions=None, force=False)#
Function for downloading model from directory.
- otx.core.ov.omz_wrapper.get_model_configuration(model_name)#
Getter function of model configuration from name.
- otx.core.ov.omz_wrapper.get_omz_model(model_name, download_dir='/home/runner/.cache/otx/omz', output_dir='/home/runner/.cache/otx/omz', force=False)#
Get OMZ model from name and download_dir.
Registry Class for otx.core.ov.
- class otx.core.ov.registry.Registry(name, add_name_as_attr=False)#
Registry Class for OMZ model.
- REGISTERED_NAME_ATTR = '_registered_name'#
- get(key: Any) Any #
Get from module name (key).
- register(name: Optional[Any] = None)#
Register from name.
- property registry_dict: Dict[Any, Any]#
Dictionary of registered module.
Utils for otx.core.ov.
- otx.core.ov.utils.get_op_name(op_node: Node) str #
Get op name string.
- otx.core.ov.utils.load_ov_model(model_path: str, weight_path: Optional[str] = None, convert_dynamic: bool = False) Model #
Load ov_model from model_path.
- otx.core.ov.utils.normalize_name(name: str) str #
Normalize name string.
- otx.core.ov.utils.to_dynamic_model(ov_model: Model) Model #
Convert ov_model to dynamic Model.
- otx.core.ov.utils.unnormalize_name(name: str) str #
Unnormalize name string.