将Tensorflow .Pb转换为ONNX [英] Tensorflow .Pb to 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>
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 share, explore and talk to experts about Microsoft Kaizala.
这篇关于将Tensorflow .Pb转换为ONNX的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!