我目前正在尝试将我使用本教程 (https://github.com/thtrieu/darkflow) 创建的已保存（和工作）.pb 文件转换为 onnx 文件。我目前正在使用winML工具，但是转换的结果根本不起作用（输入参数错误+整个架构不正确）。
我的输入参数（在自述文件的最底部指定）：输入：0
输出节点：输出：0
我想在 HoloLens 上运行的 UWP 应用程序中使用转换后的模型。
是否有人已经成功将此模型（或任何 TensorFlow 模型）转换为 ONNX？如果是这样，我将非常感谢任何链接/教程/帮助！
如果您需要更多信息，请告诉我。我对这样的论坛非常陌生。对此感到抱歉。
沃尔特
【问题讨论】：
标签： python tensorflow uwp darkflow onnx
您可以使用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 应用程序来验证输入和输出名称。
【讨论】：