转换onnx_ONNX简化节点和修改教程

在训练完深度学习的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简化器,然后选择要简化的模型即可。

转换onnx_ONNX简化节点和修改教程_第1张图片

方法2:手动修改节点

你可能感兴趣的:(转换onnx)