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