parse_metadata_from_onnx
parse_metadata_from_onnx(onnx_path: str | Path) -> dict[str, Any]
-
説明:ONNX の
custom_metadata_mapを読み取り、文字列フィールドに対してjson.loads()を試みて元の型へ戻します。 -
依存関係
onnxruntimeが必要です(InferenceSession経由で metadata を読み取ります)。
-
パラメータ
- onnx_path (
str | Path):ONNX モデルのパス。
- onnx_path (
-
戻り値
- dict[str, Any]:パース済み metadata。
-
例
from capybara.onnxengine import parse_metadata_from_onnx
meta = parse_metadata_from_onnx("model.onnx")
print(meta)