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

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

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

onnxsim-让导出的onnx模型更精简_第1张图片

        使用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、模型可视化查看,结果导出如下:

onnxsim-让导出的onnx模型更精简_第2张图片

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

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

你可能感兴趣的:(未分类,可视化,python)