onnx.hub¶

模型信息¶

class onnx.hub.ModelInfo(raw_model_info: dict[str, Any])[来源]¶

一个类，用于表示ONNX Hub中模型的属性和元数据。它从传入的raw_model_info字典中提取模型名称、路径、sha、标签等。

model¶: 模型的名称。

model_path¶: 模型的路径，相对于模型库（https://github.com/onnx/models/）仓库根目录。

metadata¶: 模型的附加元数据，例如模型的大小、IO端口等。

model_sha¶: 模型文件的SHA256摘要。

tags¶: 与模型关联的一组标签。

opset¶: 模型的opset版本。

下载带有测试数据的模型¶

onnx.hub.download_model_with_test_data(model: str, repo: str = 'onnx/models:main', opset: int | None = None, force_reload: bool = False, silent: bool = False) → str | None[source]¶

从onnx模型中心按名称下载模型以及测试数据，并返回文件解压到的目录。用户需确保模型来自可信来源，并且数据可以安全解压。

Parameters:

model – 清单中onnx模型的名称。此字段区分大小写
repo – 模型仓库的位置，格式为“用户/仓库[:分支]”。如果未找到分支，则默认为“main”。
opset – 要下载的模型的操作集。默认值为None，自动选择最大的操作集
force_reload – 是否强制模型重新下载，即使它已经在缓存中找到
silent – 如果仓库不受信任，是否抑制警告消息。

Returns:

字符串或无

获取模型信息¶

onnx.hub.get_model_info(model: str, repo: str = 'onnx/models:main', opset: int | None = None) → ModelInfo[source]¶

获取与给定名称和操作集匹配的模型信息。

Parameters:

model – 清单中onnx模型的名称。此字段区分大小写
repo – 模型仓库的位置，格式为“用户/仓库[:分支]”。如果未找到分支，则默认为“main”
opset – 要获取的模型的操作集。默认值为None将返回具有最大操作集的模型。

Returns:

ModelInfo.

列出模型¶

onnx.hub.list_models(repo: str = 'onnx/models:main', model: str | None = None, tags: list[str] | None = None) → list[ModelInfo][source]¶

获取与给定名称和标签一致的模型信息列表

Parameters:

repo – 模型仓库的位置，格式为“用户/仓库[:分支]”。如果未找到分支，则默认为“main”。
model – 要搜索的模型名称。如果为None，将返回所有具有匹配标签的模型。
tags – 用于过滤模型的标签列表。如果为None，将返回所有名称匹配的模型。

Returns:

``ModelInfo``s.

加载¶

onnx.hub.load(model: str, repo: str = 'onnx/models:main', opset: int | None = None, force_reload: bool = False, silent: bool = False) → ModelProto | None[源代码]¶

从onnx模型中心按名称下载模型。

Parameters:

model – 清单中onnx模型的名称。此字段区分大小写
repo – 模型仓库的位置，格式为“用户/仓库[:分支]”。如果未找到分支，则默认为“main”
opset – 要下载的模型的操作集。默认值为None，自动选择最大的操作集
force_reload – 是否强制模型重新下载，即使它已经在缓存中找到
silent – 如果仓库不受信任，是否抑制警告消息。

Returns:

ModelProto 或无

加载复合模型¶

onnx.hub.load_composite_model(network_model: str, preprocessing_model: str, network_repo: str = 'onnx/models:main', preprocessing_repo: str = 'onnx/models:main', opset: int | None = None, force_reload: bool = False, silent: bool = False) → ModelProto | None[source]¶

通过下载网络和预处理模型并组合成一个单一模型，构建一个包括数据预处理的复合模型

Parameters:

network_model – 清单中onnx模型的名称。
preprocessing_model – 预处理模型的名称。
network_repo – 模型仓库的位置，格式为“用户/仓库[:分支]”。如果未找到分支，则默认为“main”。
preprocessing_repo – 预处理模型仓库的位置，格式为“用户/仓库[:分支]”。如果未找到分支，则默认为“main”。
opset – 要下载的模型的操作集。默认值为None，自动选择最大的操作集
force_reload – 是否强制模型重新下载，即使它已经在缓存中找到
silent – 如果仓库不受信任，是否抑制警告消息。

Returns:

ModelProto 或无

设置目录¶

onnx.hub.set_dir(new_dir: str) → None[source]¶

设置当前的ONNX中心缓存位置。

Parameters:: new_dir – 新模型中心缓存的位置。

获取目录¶

onnx.hub.get_dir() → str[source]¶

获取当前的ONNX中心缓存位置。

Returns:: ONNX 中心模型缓存的位置。