Python将ONNX转为Json脚本

 Python脚本

import onnx
from onnx.shape_inference import infer_shapes
import numpy as npfrom google.protobuf.json_format import MessageToJson, Parse
import argparse
import osdef convertToJson(onnx_model_path):onnx_model = onnx.load(onnx_model_path)message = MessageToJson(onnx_model)dir_path, model_name = os.path.split(onnx_model_path)json_model_name = model_name.replace('.onnx', '.json')json_model_path = dir_path + "/" + json_model_namewith open(json_model_path, "w") as fo:fo.write(message)print("Save json to ", json_model_path)def parseArgs():parser = argparse.ArgumentParser(description='ONNX model to json')parser.add_argument("--onnx_model", type=str, required=True, help="Root path of ONNX model.")args = parser.parse_args()print("ONNX model: &