get_onnx_metadata
-
依賴
- 此函式需要
onnxruntime(透過 InferenceSession 讀取 metadata)。
- 此函式需要
-
說明:從 ONNX 模型中取得自定義元數據。
-
參數
- onnx_path (
Union[str, Path]):ONNX 模型的路徑。
- onnx_path (
-
傳回值
- dict[str, Any]:自定義元數據(原始
custom_metadata_map內容,通常是字串)。
- dict[str, Any]:自定義元數據(原始
-
範例
from capybara.onnxengine import get_onnx_metadata
onnx_path = 'model.onnx'
metadata = get_onnx_metadata(onnx_path)
print(metadata)
# >>> metadata = {
# 'key1': 'value1',
# 'key2': 'value2',
# 'key3': 'value3',
# }