write_metadata_into_onnx
-
依存関係
onnxが必要です(モデルファイルの read/write)。drop_old_meta=Falseの場合、旧 metadata を読み取るためにonnxruntimeも必要です。
-
説明:custom metadata を ONNX モデルへ書き込みます。
-
パラメータ
- onnx_path (
str | Path):ONNX モデルのパス。 - out_path (
str | Path):出力 ONNX モデルのパス。 - drop_old_meta (
bool):旧 metadata を削除するかどうか。デフォルトはFalse。 **kwargs:書き込む custom metadata。
- onnx_path (
-
挙動
Dateフィールドを自動追加します(capybara.utils.time.now(fmt=...)を使用)。- 各 metadata value は
json.dumps()で文字列化して ONNX props に書き込みます。
-
例
from capybara.onnxengine import write_metadata_into_onnx
onnx_path = 'model.onnx'
out_path = 'model_with_metadata.onnx'
write_metadata_into_onnx(
onnx_path,
out_path,
drop_old_meta=False,
key1='value1',
key2='value2',
key3='value3',
)