2

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

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

输出节点:输出:0

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

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

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

沃尔特

4

2 回答 2

3

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

此外,tf2onnx可能很有用

于 2019-07-09T07:54:53.557 回答
2

您可以使用tf2onnx将您的 .pb 文件转换为 onnx。

使用 pip 安装它。

pip install tf2onnx

然后你这样称呼它。

python -m tf2onnx.convert --input <path to your pb file> --inputs <comma-delimited input names goes here> --outputs <comma-delimited output names goes here> --output <path to output in onnx format>

你说你的输入是input:0,你的输出是ouput:0,所以命令应该是这样的。

python -m tf2onnx.convert --input yolo.pb --inputs input:0 --outputs ouput:0 --output yolo.onnx

如有必要,您可以使用netron应用程序来验证输入和输出名称。

于 2021-12-08T18:31:22.940 回答