安装onnxsim

安装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')

step3、模型可视化查看,结果导出如下:
在这里插入图片描述
在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/qq_35037684/article/details/121248694