将Tensorflow .Pb转换为ONNX [英] Tensorflow .Pb to ONNX

查看:3802
本文介绍了将Tensorflow .Pb转换为ONNX的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!

问题描述

我已经在Azure ML服务中构建了一个Tensorflow模型.我将模型保存为* .pb格式.为了启用预测,我希望将其转换为ONNX格式.谁能建议这些步骤或为我提供示例.

I have built a tensorflow model in Azure ML service. I got the model saved in *.pb format. To enable predictions I want to get this converted to ONNX format. Can anyone suggest the steps or point me to an example.

谢谢

推荐答案

您可以使用github cmd界面,让我知道您需要任何帮助.< o:p></o:p>

https://github.com/onnx/onnx-tensorflow<o:p></o:p>


从Tensorflow到ONNX:

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

从ONNX到Tensorflow:

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

Hope this helps...

---------------------------------------------------------------------------------------------------------

"他们提供了帮助,还设置了" "" ; 这对于我们继续写博客 J

Please don't forget to "mark the replies as answers" if they helped, also set "like" it’s a boost for us to keep blogging J

学到更多.访问专用的 & 有关Microsoft Kaizala的专家.

Click here to learn more. Visit the dedicated Community forum to shareexplore and talk to experts about Microsoft Kaizala.


这篇关于将Tensorflow .Pb转换为ONNX的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!

查看全文
登录 关闭
扫码关注1秒登录
发送“验证码”获取 | 15天全站免登陆