get_onnx_metadata
-
說明:從 ONNX 模型中取得自定義元數據。
-
參數
- onnx_path (
Union[str, Path]
):ONNX 模型的路徑。
- onnx_path (
-
傳回值
- dict:自定義元數據。
-
範例
import capybara as cb
onnx_path = 'model.onnx'
metadata = cb.get_onnx_metadata(onnx_path)
print(metadata)
# >>> metadata = {
# 'key1': 'value1',
# 'key2': 'value2',
# 'key3': 'value3',
# }