工资「喂饱肚子」，副业「养活灵魂」！
问题反馈可发送邮件到stubbornhuang@qq.com
如果觉得本站的内容有帮助，可以考虑打赏博主哦！
计算机图形学与计算几何经典必备书单整理，下载链接可参考：https://www.stubbornhuang.com/1256/
在本站开通年度VIP，无限制下载本站资源和阅读本站文章
本站会放置Google广告用于维持域名以及网站服务器费用。
欢迎大家交换友链，可在https://www.stubbornhuang.com/申请友情链接进行友链交换申请！
本站由于前段时间遭受到大量临时和国外邮箱注册，所以对可注册的邮箱类型进行了限制！
感谢大家访问本站，希望本站的内容可以帮助到大家！

Onnx – onnx模型简化与优化

StubbornHuang Onnx 发布于2023-03-21 阅读 5,979次 0次评论 0次点赞本文共861个字，阅读需要3分钟。

在使用onnx模型在其他框架进行部署时，一般都需要对onnx模型进行简化和优化。

1 onnx模型简化

对模型进行简化一般使用onnx-simplifier包

可以通过以下命令安装

pip install onnx-simplifier

然后就可以在项目导出onnx模型之后，使用以下代码对onnx模型进行简化

import onnx
from onnxsim import simplify

if __name__ == '__main__':
    onnx_path = './mobile.onnx'
    onnx_simp_path = './mobile_simp.onnx'
    onnx_model = onnx.load(onnx_path)  # 加载onnx模型
    onnx_model_simp, check = simplify(onnx_model)  # 简化onnx模型
    assert check, "Simplified ONNX model could not be validated"
    onnx.save(onnx_model_simp, onnx_simp_path)

也可以在python环境中使用以下命令行完成简化

python3 -m onnxsim ./mobile.onnx ./mobile_simp.onnx

2 onnx模型优化

对模型进行简化一般使用onnxoptimizer包

可以通过以下命令安装

pip install onnxoptimizer

使用以下代码对onnx模型进行优化

import onnx
import onnxoptimizer

if __name__ == '__main__':
    onnx_path = './mobile.onnx'
    onnx_optim_path = './mobile_optim.onnx'
    onnx_model = onnx.load(onnx_path)  # 加载onnx模型
    onnx_model_optim = onnxoptimizer.optimize(onnx_model)  # 优化onnx模型
    onnx.save(onnx_model_optim, onnx_optim_path)

欢迎扫码关注我的微信公众号，及时获取文章更新