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)