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 到 Tensorflowonnx-tf convert -t tf -i /path/to/input.onnx -o /path/to/output.pb

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

 

  • 2
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 23
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 23
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值