onnx和json互相转换

json2onnx

import onnx
from onnx.shape_inference import infer_shapes
import tflite2onnx
import numpy as np
 
import onnx
from google.protobuf.json_format import MessageToJson, Parse
import json
with open("conv.json", "r") as fi:
    onnx_json = json.loads(fi.read())
    onnx_str = json.dumps(onnx_json)
    convert_model = Parse(onnx_str, onnx.ModelProto())
    onnx.save(convert_model, "Conv_1.onnx")

onnx2json

import onnx
from onnx.shape_inference import infer_shapes
import tflite2onnx
import numpy as np
 
import onnx
from google.protobuf.json_format import MessageToJson, Parse
 
onnx_model = onnx.load("/home/wyf/3gi/test/rknntest1/rk-test-1808/onnx_change_input/model_640x384.onnx")
message = MessageToJson(onnx_model)
with open("conv.json", "w") as fo:
    fo.write(message)

更多内容请关注公众号，了解嵌入式ai前沿资讯！