我目前正在尝试将我使用本教程创建的已保存(且正在工作)的 .pb 文件 ( https://github.com/thtrieu/darkflow ) 转换为 onnx 文件。我现在正在使用winML工具,但是转换的结果根本不起作用(输入参数错误+整个架构不正确)。
我的输入参数(如自述文件最底部指定):input:0
输出节点:ouput:0
我想在 HoloLens 上运行的 UWP 应用程序中使用转换后的模型。
是否有人已成功将此模型(或任何 TensorFlow 模型)转换为 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应用程序验证输入和输出名称。
关于python - Tensorflow 到 ONNX 的转换,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54214470/