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.