python - Tensorflow 到 ONNX 的转换

Question

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

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

输出节点:ouput:0

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

是否有人已成功将此模型(或任何 TensorFlow 模型)转换为 ONNX？

Answer 1

您可以使用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应用程序验证输入和输出名称。