本文作者:StubbornHuang
版权声明:本文为站长原创文章,如果转载请注明原文链接!
原文标题:Onnx – onnx模型简化与优化
原文链接:https://www.stubbornhuang.com/2555/
发布于:2023年03月21日 15:50:28
修改于:2023年03月21日 15:50:28
在使用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)
当前分类随机文章推荐
- Onnx - onnx模型简化与优化 阅读139次,点赞0次
全站随机文章推荐
- Windows - OneDrive登录出现0x8019001错误 阅读343次,点赞0次
- WordPress - 禁止非管理员登录后台 阅读1778次,点赞0次
- 资源分享 - OpenGL编程指南(原书第7版)- OpenGL红宝书高清扫描版PDF下载 阅读4886次,点赞1次
- C++11 - override关键字简要介绍 阅读2132次,点赞0次
- OnnxRuntime – 模型部署笔记2,在Visual Studio中配置Onnx Rumtime C++开发环境 阅读201次,点赞0次
- UnrealEngine4 - 将TextureRenderTarget2D保存为图片 阅读2091次,点赞0次
- Pytorch - 使用pytorch自带的Resnet作为网络的backbone 阅读353次,点赞0次
- 资源分享 - Geometry for Computer Graphics - Formulae, Examples and Proofs 英文高清PDF下载 阅读1510次,点赞0次
- 资源分享 - Handbook of Computer Aided Geometric Design 英文高清PDF下载 阅读1809次,点赞0次
- C++ – 字节数组byte[]或者unsigned char[]与long long的相互转换 阅读1662次,点赞0次
评论
169