首页 > 解决方案 > TensorFlow 到 ONNX 的转换

问题描述

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

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

输出节点:输出:0

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

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

如果您需要更多信息,请告诉我。我对这样的论坛非常陌生。对于那个很抱歉。

沃尔特

标签: pythontensorflowuwpdarkflowonnx

解决方案


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

此外,tf2onnx可能很有用


推荐阅读