onnx模型转tensorflow模型

版权声明:热爱生活,热爱分享 https://blog.csdn.net/wcx1293296315/article/details/88699534

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

1、安装tensorflow和onnx

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

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

2、安装onnx-tensorflow

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

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

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

3、onnx转tensorflow

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

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

猜你喜欢

转载自blog.csdn.net/wcx1293296315/article/details/88699534