Tensorflow到ONNX的转换

时间:2019-01-16 09:57:13

标签: python tensorflow uwp darkflow onnx

我目前正在尝试将通过本教程(https://github.com/thtrieu/darkflow)创建的已保存(并且正在工作)的.pb文件转换为onnx文件。我目前正在使用winML工具,但是转换的结果根本不起作用(输入参数错误+整个体系结构不正确)。

我的输入参数(在自述文件的最底部指定):input:0

输出节点:输出:0

我想在HoloLens上运行的UWP应用程序中使用转换后的模型。

有人已经成功将此模型(或任何TensorFlow模型)转换为ONNX吗?如果是这样,我将非常感谢您提供的任何链接/教程/帮助!

如果您需要更多信息,请告诉我。我对此类论坛没有任何经验。抱歉。

沃尔特

1 个答案:

答案 0 :(得分:0)

我用MMdnn将多个模型从TF转换为onnx

另外,tf2onnx可能有用