io_helper

onnx_extended.onnx2.load(f: str | Path, skip_raw_data: bool = False, raw_data_threshold: int = 1024, load_external_data: bool = True) → ModelProto[source]

Loads a serialized ModelProto into memory.

Parameters:

  • f – path or bytes

  • skip_raw_data – skips the raw data of every tensor, this can be used to load only the architecture of the model even if the model is stored in one unique file

  • raw_data_threshold – if skip_raw_data is True, still keeps the tensors smaller than this size (in bytes)

  • load_external_data – Whether to load the external data. Set to True if the data is under the same directory of the model.

Returns:

Loaded in-memory ModelProto.

onnx_extended.onnx2.save(proto: ModelProto, f: str | Path, format: str = 'protobuf', *, save_as_external_data: bool = False, all_tensors_to_one_file: bool = True, location: str | None = None, size_threshold: int = 1024, convert_attribute: bool = False) → None[source]

Saves the ModelProto to the specified path and optionally, serializes tensors with raw data as external data before saving.

Parameters:

  • proto – should be a in-memory ModelProto

  • f – can be a file-like object (has “write” function) or a string containing a file name or a pathlike object

  • format – The serialization format. When it is not specified, it is inferred from the file extension when f is a path. If not specified _and_ f is not a path, ‘protobuf’ is used. The encoding is assumed to be “utf-8” when the format is a text format.

  • save_as_external_data – If true, save tensors to external file(s). all_tensors_to_one_file: Effective only if save_as_external_data is True. If true, save all tensors to one external file specified by location. If false, save each tensor to a file named with the tensor name.

  • all_tensors_to_one_file – if save_as_external_data is True, then saves all tensors into one file instead of a file per tensor

  • location – Effective only if save_as_external_data is true. Specify the external file that all tensors to save to. Path is relative to the model path. If not specified, will use the model name.

  • size_threshold – Effective only if save_as_external_data is True. Threshold for size of data. Only when tensor’s data is >= the size_threshold it will be converted to external data. To convert every tensor with raw data to external data set size_threshold=0.

  • convert_attribute – Effective only if save_as_external_data is True. If true, convert all tensors to external data If false, convert only non-attribute tensors to external data