onnx.hub¶
ModelInfo¶
- 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 版本。
download_model_with_test_data¶
- 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[來源]¶
從 onnx 模型中心下載具有測試資料的模型，並傳回已將檔案解壓縮到的目錄。使用者有責任確保模型來自可信任的來源，並且資料可以安全地解壓縮。
- 參數:
model – 清單中 onnx 模型的名稱。此欄位區分大小寫
repo – 模型儲存庫的位置，格式為「使用者/儲存庫[:分支]」。如果找不到任何分支，則預設為「main」
opset – 要下載模型的 opset。預設值 None 會自動選擇最大的 opset
force_reload – 是否強制重新下載模型，即使它已在快取中找到
silent – 如果儲存庫不受信任，是否抑制警告訊息。
- 傳回:
str 或 None
get_model_info¶
list_models¶
- onnx.hub.list_models(repo: str = 'onnx/models:main', model: str | None = None, tags: list[str] | None = None) list[ModelInfo][原始碼]¶
取得符合指定名稱和標籤的模型資訊列表
- 參數:
repo – 模型儲存庫的位置，格式為「使用者/儲存庫[:分支]」。如果找不到任何分支，則預設為「main」
model – 要搜尋的模型名稱。如果為 None，將會回傳所有符合標籤的模型。
tags – 用來篩選模型的標籤列表。如果為 None，將會回傳所有符合名稱的模型。
- 傳回:
``ModelInfo``s。
load¶
- onnx.hub.load(model: str, repo: str = 'onnx/models:main', opset: int | None = None, force_reload: bool = False, silent: bool = False) ModelProto | None[原始碼]¶
從 onnx 模型中心以下載名稱載入模型。
- 參數:
model – 清單中 onnx 模型的名稱。此欄位區分大小寫
repo – 模型儲存庫的位置，格式為「使用者/儲存庫[:分支]」。如果找不到任何分支，則預設為「main」
opset – 要下載模型的 opset。預設值 None 會自動選擇最大的 opset
force_reload – 是否強制重新下載模型，即使它已在快取中找到
silent – 如果儲存庫不受信任，是否抑制警告訊息。
- 傳回:
ModelProto 或 None
load_composite_model¶
- 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[原始碼]¶
透過下載網路模型和預處理模型並將其組合到單一模型中，建立包含資料預處理的複合模型
- 參數:
network_model – 清單中 onnx 模型的名稱。
preprocessing_model – 預處理模型的名稱。
network_repo – 模型存放庫的位置，格式為「使用者/存放庫[:分支]」。如果找不到分支，則預設為「main」
preprocessing_repo – 預處理模型存放庫的位置，格式為「使用者/存放庫[:分支]」。如果找不到分支，則預設為「main」
opset – 要下載模型的 opset。預設值 None 會自動選擇最大的 opset
force_reload – 是否強制重新下載模型，即使它已在快取中找到
silent – 如果儲存庫不受信任，是否抑制警告訊息。
- 傳回:
ModelProto 或 None