Models#

Module for otx.core.ov.models.

class otx.core.ov.models.MMOVModel(model_path_or_model: Union[str, Model], weight_path: Optional[str] = None, inputs: Optional[Union[Dict[str, Union[str, List[str]]], List[str], str]] = None, outputs: Optional[Union[Dict[str, Union[str, List[str]]], List[str], str]] = None, *args, **kwargs)#

MMOVModel for OMZ model type.

forward(inputs, gt_label=None)#: Function forward.

training: bool#

class otx.core.ov.models.OVModel(model_path_or_model: Optional[Union[str, Model]] = None, weight_path: Optional[str] = None, inputs: Optional[Union[str, List[str]]] = None, outputs: Optional[Union[str, List[str]]] = None, features_to_keep: Optional[List] = None, remove_normalize: bool = False, merge_bn: bool = True, paired_bn: bool = True, init_weight: Union[bool, Callable] = False, verify_shape: bool = True)#

OVModel class.

static build_custom_inputs(graph, inputs: Union[str, List[str]])#: Function build_custom_inputs.

static build_custom_outputs(graph, outputs)#: Function build_custom_outputs.

static build_graph(model_path_or_model, weight_path=None)#: Function build_graph.

static build_torch_module(graph)#: Function build_torch_module.

static clean_up(graph, inputs=None, outputs=None)#: Function clean_up.

property features#: Property features.

forward(*args, **kwargs)#: Function forward.

property input_shapes#: Property input_shapes.

property inputs#: Property inputs.

property output_shapes#: Property output_shapes.

property outputs#: Property outputs.

training: bool#

class otx.core.ov.models.ParserMixin#

ParserMixin class.

parse(model_path_or_model: Union[str, Model], weight_path: Optional[str] = None, inputs: Optional[Union[Dict[str, Union[str, List[str]]], List[str], str]] = None, outputs: Optional[Union[Dict[str, Union[str, List[str]]], List[str], str]] = None, parser: Optional[Union[str, Callable]] = None, **kwargs) → Tuple[Union[str, List[str]], Union[str, List[str]]]#: Parse function of ParserMixin class.

static parser(graph, **kwargs) → Dict[str, Union[List[str], Dict[str, List[str]]]]#: Function parser.

MMOVModel for otx.core.ov.models.mmov_model.

class otx.core.ov.models.mmov_model.MMOVModel(model_path_or_model: Union[str, Model], weight_path: Optional[str] = None, inputs: Optional[Union[Dict[str, Union[str, List[str]]], List[str], str]] = None, outputs: Optional[Union[Dict[str, Union[str, List[str]]], List[str], str]] = None, *args, **kwargs)#

MMOVModel for OMZ model type.

forward(inputs, gt_label=None)#: Function forward.

training: bool#

Modules for otx.core.ov.models.ov_model.

class otx.core.ov.models.ov_model.OVModel(model_path_or_model: Optional[Union[str, Model]] = None, weight_path: Optional[str] = None, inputs: Optional[Union[str, List[str]]] = None, outputs: Optional[Union[str, List[str]]] = None, features_to_keep: Optional[List] = None, remove_normalize: bool = False, merge_bn: bool = True, paired_bn: bool = True, init_weight: Union[bool, Callable] = False, verify_shape: bool = True)#

OVModel class.

static build_custom_inputs(graph, inputs: Union[str, List[str]])#: Function build_custom_inputs.

static build_custom_outputs(graph, outputs)#: Function build_custom_outputs.

static build_graph(model_path_or_model, weight_path=None)#: Function build_graph.

static build_torch_module(graph)#: Function build_torch_module.

static clean_up(graph, inputs=None, outputs=None)#: Function clean_up.

property features#: Property features.

forward(*args, **kwargs)#: Function forward.

property input_shapes#: Property input_shapes.

property inputs#: Property inputs.

property output_shapes#: Property output_shapes.

property outputs#: Property outputs.

training: bool#

Parser mixin modules for otx.core.ov.models.

class otx.core.ov.models.parser_mixin.ParserMixin#

ParserMixin class.

parse(model_path_or_model: Union[str, Model], weight_path: Optional[str] = None, inputs: Optional[Union[Dict[str, Union[str, List[str]]], List[str], str]] = None, outputs: Optional[Union[Dict[str, Union[str, List[str]]], List[str], str]] = None, parser: Optional[Union[str, Callable]] = None, **kwargs) → Tuple[Union[str, List[str]], Union[str, List[str]]]#: Parse function of ParserMixin class.

static parser(graph, **kwargs) → Dict[str, Union[List[str], Dict[str, List[str]]]]#: Function parser.