onnx转onnxsim

最新推荐文章于 2024-09-13 07:39:25 发布

KongTiaoXuLun

最新推荐文章于 2024-09-13 07:39:25 发布

阅读量258

点赞数 2

文章标签： python linux 开发语言

本文链接：https://blog.csdn.net/cyh20182808/article/details/141861717

版权

参考链接：

onnx转onnxsim-CSDN博客

简化onnx模型

先在环境里安装

 pip install onnx-simplifier

然后简化的代码是

import onnx
from onnxsim import simplify
 
ONNX_MODEL_PATH = './model.onnx'#需要被简化的onnx
ONNX_SIM_MODEL_PATH = './model_simple.onnx'#简化后的onnx保存
 
if __name__ == "__main__":
    onnx_model = onnx.load(ONNX_MODEL_PATH)
    onnx_sim_model, check = simplify(onnx_model)
    assert check, "Simplified ONNX model could not be validated"
    onnx.save(onnx_sim_model, ONNX_SIM_MODEL_PATH)
    print('ONNX file simplified!')

就生成了