hubai_sdk.utils.metadata

module documentation

Read tensor shapes and element types from supported model files.

Class	`Metadata`	Input and output tensor metadata extracted from a model.
Function	`get_metadata`	Extract input and output metadata from a supported model file.
Function	`_get_metadata_ir`	Extract metadata from an OpenVINO IR model.
Function	`_get_metadata_onnx`	Extract metadata from an ONNX model.
Function	`_get_metadata_tflite`	Extract metadata from a TFLite model.

def get_metadata(model_path: Path) -> Metadata: ¶

Extract input and output metadata from a supported model file.

Parameters
model_path:`Path`	Path to an ONNX, OpenVINO IR, or TFLite model.

Returns
`Metadata`	Input and output shapes and dtypes for the model.

Raises
`ValueError`	If the model format is unsupported.

def _get_metadata_ir(bin_path: Path, xml_path: Path) -> Metadata: ¶

Extract metadata from an OpenVINO IR model.

def _get_metadata_onnx(onnx_path: Path) -> Metadata: ¶

Extract metadata from an ONNX model.

def _get_metadata_tflite(model_path: Path) -> Metadata: ¶

Extract metadata from a TFLite model.