onnx模型转tensorflow模型

ONNX是开源神经网络交换平台，有了它基本上不用纠结用什么深度学习框架的问题了。我现在记录一下怎么将onnx模型转换成tensorflow模型。

1、安装tensorflow和onnx

我是通过Anaconda安装的。【详情】这个博客记载了安装Anaconda和onnx的详情，安装好Anaconda后，至于安装tensorflow只需要输入【conda install tensorflow】就行了。

注意：如果已经通过Anaconda安装过其他深度模型框架，建议再创建一个新环境，否则安装tensorflow时，其他深度学习框架将被删除。反正我原先安装的caffe2以及onnx，在安装tensorflow后都就不见了。

2、安装onnx-tensorflow

$ git clone git@github.com:onnx/onnx-tensorflow.git && cd onnx-tensorflow ###通过git安装onnx-tensorflow模块 $ pip install -e . ###注意那个点也是需要的

通过git进行拉取时，你可能会遇到permission错误，这个不是能通过sudo解决的。因为这个错误的原因不在这，欲知详情请到这个博客中查看。

安装好后，试一下【onnx-tf -h】这个命令可不可以用。没报错就行了

3、onnx转tensorflow

从 ONNX 到 Tensorflow: onnx-tf convert -t tf -i /path/to/input.onnx -o /path/to/output.pb

反正现在看起来是转成功了。哈哈哈哈哈哈

 

总结

以上是生活随笔为你收集整理的onnx模型转tensorflow模型的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。