安装onnxsim

我现在强的可怕~

已于 2023-05-26 17:55:03 修改

阅读量1.7w

点赞数 13

分类专栏：深度学习文章标签： python

于 2021-11-10 15:03:35 首次发布

本文链接：https://blog.csdn.net/qq_35037684/article/details/121248694

版权

深度学习专栏收录该内容

121 篇文章

订阅专栏

这篇博客介绍了如何正确安装和使用onnx-simplifier工具来简化ONNX模型。首先，通过pip安装onnx-simplifier包。然后，加载ONNX模型，使用simplify函数进行简化，并保存。最后，提供了模型简化后的可视化方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

安装onnxsim并不是pip install onnxsim, 这样会报错

正确的安装方式：

step1、安装onnxsim包

pip install onnx-simplifier

step2、加载onnx文件，simplify处理后重新保存，代码如下：

from onnxsim import simplify
onnx_model = onnx.load(output_path)  # load onnx model
model_simp, check = simplify(onnx_model)
assert check, "Simplified ONNX model could not be validated"
onnx.save(model_simp, output_path)
print('finished exporting onnx')