用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博客