hubai_sdk.utils.config

module documentation

Undocumented

Class BlobBaseConfig Undocumented
Class Config No class docstring; 0/2 instance variable, 0/1 class variable, 2/2 methods, 0/2 class method documented
Class CustomBaseModel Undocumented
Class EncodingConfig Undocumented
Class HailoConfig Undocumented
Class InputConfig No class docstring; 0/2 property, 0/4 class variable, 0/2 method, 3/3 static methods, 0/1 class method documented
Class OutputConfig Undocumented
Class RVC2Config Undocumented
Class RVC3Config Undocumented
Class RVC4Config Undocumented
Class SingleStageConfig No class docstring; 0/15 class variable, 1/1 method, 2/2 class methods documented
Class TargetConfig Undocumented
Constant NAMED_VALUES Undocumented
Function _extract_bin_xml_from_ir Extracts the corresponding second path from a single IR path.
Function _get_onnx_inter_info Try tensor lookup first, then node lookup, for ONNX metadata.
Function _get_onnx_node_info Read shape and dtype information for an ONNX node output.
Function _get_onnx_tensor_info Read shape and dtype information for an ONNX tensor.
NAMED_VALUES: dict[str, dict] =

Undocumented

Value
{'imagenet': {'mean': [123.675, 116.28, 103.53],
              'scale': [58.395, 57.12, 57.375]}}
def _extract_bin_xml_from_ir(ir_path: Any, input_bin_path: Any | None = None) -> tuple[Path, Path]:

Extracts the corresponding second path from a single IR path.

We assume that the base filename matches between the .bin and .xml file unless an explicit input_bin_path is provided.

def _get_onnx_inter_info(model_path: Path, name: str) -> tuple[list[int] | None, DataType | None]:

Try tensor lookup first, then node lookup, for ONNX metadata.

def _get_onnx_node_info(model_path: Path, node_name: str) -> tuple[list[int], DataType]:

Read shape and dtype information for an ONNX node output.

def _get_onnx_tensor_info(model_path: Path | str, tensor_name: str) -> tuple[list[int], DataType]:

Read shape and dtype information for an ONNX tensor.