onnxsim-让导出的onnx模型更精简

2023-05-16

        用torch导出的模型有时候参数过多,不利于查看。查找资料onnxsim可以简化模型,让显示更加自然。

        如,reshap层的导出,红框中细节参数被显示出现。默认导出:

        使用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')

step3、模型可视化查看,结果导出如下:

        是不是更精简了呢。可视化可使用Netron打开显示。

        更多Netron的使用方式可参考:Netron-模型结构可视化工具(支持tf, caffe, keras,mxnet,onnx,darknet)_alex1801的博客-CSDN博客

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

onnxsim-让导出的onnx模型更精简 的相关文章

随机推荐