onnx 模型转换opset version

最新推荐文章于 2024-06-24 14:08:28 发布

_Woody

最新推荐文章于 2024-06-24 14:08:28 发布

阅读量6.4k

点赞数 5

文章标签：深度学习神经网络人工智能

本文链接：https://blog.csdn.net/u014027643/article/details/126667336

版权

import onnx
from onnx import version_converter, helper

# import onnxruntime
# help(onnx)

# Preprocessing: load the model to be converted.
model_path = 'old.onnx'
original_model = onnx.load(model_path)
original_model.opset_import[0].version = 11
original_model.ir_version = 6

onnx.save(original_model, "new.onnx")
# print(original_model.Version)


# print('The model before conversion:\n{}'.format(original_model))