datumaro.plugins.openvino_plugin.launcher

Classes

BuiltinOpenvinoModelInfo(interpreter, ...)
OpenvinoLauncher([description, weights, ...])
OpenvinoModelInfo(interpreter, description, ...)

class datumaro.plugins.openvino_plugin.launcher.OpenvinoModelInfo(interpreter: str | None, description: str | None, weights: str | None, model_dir: str | None)

Bases: object

interpreter: str | None

description: str | None

weights: str | None

model_dir: str | None

validate()

Validate integrity of the member variables

class datumaro.plugins.openvino_plugin.launcher.BuiltinOpenvinoModelInfo(interpreter: str | None, description: str | None, weights: str | None, model_dir: str | None)

Bases: OpenvinoModelInfo

downloadable_models = {'clip_text_ViT-B_32', 'clip_text_vit_l_14_336px_int8', 'clip_visual_ViT-B_32', 'clip_visual_vit_l_14_336px_int8', 'googlenet-v4-tf'}

classmethod create_from_model_name(model_name: str) → BuiltinOpenvinoModelInfo

override(other: OpenvinoModelInfo) → None

Override builtin model variables to other

class datumaro.plugins.openvino_plugin.launcher.OpenvinoLauncher(description: str | None = None, weights: str | None = None, interpreter: str | None = None, model_dir: str | None = None, model_name: str | None = None, output_layers: List[str] = [], device: str | None = None, compile_model_config: Dict | None = None)

Bases: LauncherWithModelInterpreter

cli_plugin

alias of _OpenvinoImporter

property inputs

property outputs

infer(inputs: ndarray | Dict[str, ndarray]) → List[Dict[str, ndarray] | List[Dict[str, ndarray]]]