onnx的简化与优化

一、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

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容