一、onnx简化onnxsim
step1、安装onnxsim包
pip install onnx-simplifier
step2、加载onnx文件,simplify处理后重新保存,代码如下:
import onnx
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)
二、onnx的优化
以往使用 onnx optimizer 的方式是
import onnx
# ...
new_model = onnx.optimizer.optimize(model)
现在 onnx optimizer 在独立的仓库维护,有了自己的 onnxoptimizer
包:
pip install onnxoptimizer
import onnxoptimizer
# ...
new_model = onnxoptimizer.optimize(model)
而原 onnx
包里的 optimizer
部分在下一个版本就会删除掉。
参考:https://zhuanlan.zhihu.com/p/350702340