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
---------------------
作者:我是谁_谁是我
来源:优快云
原文:https://blog.youkuaiyun.com/wcx1293296315/article/details/88699534
版权声明:本文为博主原创文章,转载请附上博文链接!