Tensorflow 到 ONNX 的转换 [英] Tensorflow to ONNX conversion

问题描述

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

I'm currently trying to convert a saved (and working) .pb file that I created with this tutorial (https://github.com/thtrieu/darkflow) into a onnx file. I'm using winML tools at the moment but the result of the conversion doesn't work at all (the input parameters are wrong + the whole architecture is not correct).

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

My input parameter (as specified on the very bottom of the readme): input:0

输出节点:输出:0

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

I want to use the converted model inside a UWP application that's running on a HoloLens.

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

Has anyone already successfully converted this model (or any TensorFlow model) to ONNX? If so, I'd be very thankful for any link/tutorial/help!!

如果您需要更多信息，请告诉我.我对这样的论坛非常缺乏经验.很抱歉.

Please let me know if you need any more Information. I'm very unexperienced with such forums. Sorry about that.

沃尔特

推荐答案

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

此外，tf2onnx 也很有用

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