onnx.external_data_helper¶
convert_model_from_external_data¶
- onnx.external_data_helper.convert_model_from_external_data(model: ModelProto) None [源代码]¶
调用此函数,将所有使用外部数据的张量设置为嵌入数据。调用此函数后,save_model 会将所有张量数据保存为嵌入数据。
- 参数:
model (ModelProto) – 要转换的模型。
convert_model_to_external_data¶
- onnx.external_data_helper.convert_model_to_external_data(model: ModelProto, all_tensors_to_one_file: bool = True, location: str | None = None, size_threshold: int = 1024, convert_attribute: bool = False) None [源代码]¶
调用此函数,将所有带有原始数据的张量设置为外部数据。此调用应在 ‘save_model’ 之前进行。调用此函数后,‘save_model’ 会将所有张量数据保存为外部数据。
- 参数:
model (ModelProto) – 要转换的模型。
all_tensors_to_one_file (bool) – 如果为 true,将所有张量保存到由 location 指定的外部文件中。如果为 false,将每个张量保存到以张量名称命名的文件中。
location – 指定相对于模型保存所有张量的外部文件。路径相对于模型路径。如果未指定,将使用模型名称。
size_threshold – 数据的大小阈值。仅当张量的数据大小 >= size_threshold 时,它才会被转换为外部数据。要将所有带原始数据的张量转换为外部数据,请设置 size_threshold=0。
convert_attribute (bool) – 如果为 true,则将所有张量转换为外部数据;如果为 false,则仅将非属性张量转换为外部数据。
- 抛出:
ValueError – 如果 location 不是相对路径。
FileExistsError – 如果 location 中已存在文件。
ExternalDataInfo¶
- class onnx.external_data_helper.ExternalDataInfo(tensor: TensorProto)[源代码]¶
load_external_data_for_model¶
- onnx.external_data_helper.load_external_data_for_model(model: ModelProto, base_dir: str) None [源代码]¶
加载外部张量到模型中
- 参数:
model – 要加载外部数据的 ModelProto
base_dir – 包含外部数据的目录
load_external_data_for_tensor¶
- onnx.external_data_helper.load_external_data_for_tensor(tensor: TensorProto, base_dir: str) None [源代码]¶
从外部文件加载张量数据。理想情况下,TensorProto 不应包含任何原始数据,但如果包含,则会被忽略。
- 参数:
tensor – TensorProto 对象。
base_dir – 包含外部数据的目录。
remove_external_data_field¶
- onnx.external_data_helper.remove_external_data_field(tensor: TensorProto, field_key: str) None [源代码]¶
从张量的 external_data 键值对存储中移除一个字段。
就地修改张量对象。
- 参数:
tensor (TensorProto) – 将从中移除值的张量对象
field_key (string) – 要移除的字段的键
save_external_data¶
- onnx.external_data_helper.save_external_data(tensor: TensorProto, base_path: str) None [源代码]¶
根据 external_data 字段中的信息将张量数据写入外部文件。
- 参数:
tensor (TensorProto) – 要序列化的张量对象
base_path – 存储张量数据的文件夹的系统路径
set_external_data¶
uses_external_data¶
- onnx.external_data_helper.uses_external_data(tensor: TensorProto) bool [源代码]¶
如果张量数据存储在外部位置,则返回 true。
write_external_data_tensors¶
- onnx.external_data_helper.write_external_data_tensors(model: ModelProto, filepath: str) ModelProto [源代码]¶
序列化所有数据位置设置为 TensorProto.External 的张量的数据。
注意:此函数还会从所有张量的 external_data 字段中移除 basepath 信息。
- 参数:
model (ModelProto) – 要序列化的张量的来源模型对象。
filepath – 应作为外部数据基础路径的目录的系统路径。
- 返回值:
修改后的模型对象。
- 返回类型: