How to adjust a Neural Networks metadata

From Tygron Preview Support Wiki
Revision as of 12:39, 19 December 2024 by Frank@tygron.nl (talk | contribs)
Jump to navigation Jump to search
import onnx
onnx_model = onnx.load(onnxModelName)

Add a metadata key (without duplication):

def addMeta(onnx_model, key, value):
    
    for entry in onnx_model.metadata_props:
        if entry.key == key:
            entry.value = value
            return

    meta = onnx_model.metadata_props.add()
    meta.key = key
    meta.value = value  

Save the onnx model to file:

with open(onnxFilePath, "wb") as f:
    f.write(onnx_model.SerializeToString())