ONNX简化节点和修改教程

本文介绍了如何将训练好的PyTorch或TensorFlow深度学习模型转换为ONNX格式,并进行简化。推荐使用onnx-simplifier工具进行模型精简,通过pip安装后运行命令即可完成转换。此外,还提供了一个在线转换网站方便使用。对于仍有冗余节点如cast的情况,可以通过手动修改为identity节点后再进行简化。
摘要由CSDN通过智能技术生成

在训练完深度学习的pytorch或者tensorflow模型后,有时候需要把模型转成 onnx,但是很多时候,很多节点比如cast节点,Identity 这些节点可能都不需要,我们需要进行简化,这样会方便我们把模型转成ncnn或者mnn等这些端侧部署的模型格式或者通过tensorRT进行部署。

方法1:使用 onnx-simplifier 进行精简

一个方法是使用 [onnx-simplifier](https://github.com/daquexian/onnx-simplifier)  进行精简。

安装其实很很简单,使用pip进行安装:

pip3 install onnx-simplifier

```

然后进行转换:

python3 -m onnxsim input_onnx_model output_onnx_model

另外,ONNX Simplifier 还提供了在线转换的网站,可以不必安装环境就能转换,[地址戳此](https://convertmodel.com)。

开箱即用,不需要任何安装。只需打开网页,选择ONNX作为输出格式,选择onnx简化器,然后选择要简化的模型即可。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Mrrunsen

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值