我有一个TensorFlow图冻结为.pb，我想将其转换为.onnx格式。我目前正在尝试使用mmconvert（来自Microsoft的MMdnn），显然我在做错事（请参见下面的代码）。是否有其他建议的方法可以将.pb转换为.onnx？谢谢。
这是我正在使用的命令：
mmconvert -sf tensorflow -iw /frozen_models/model.pb --inNodeName input --inputShape 512 -df onnx -om tf_mobilenet --inNodeName input
这给出了错误
ValueError: Need to provide the output node of Tensorflow model.
最佳答案
Onnx格式要求在模型中指定输出节点。快速浏览一下，mmconvert期望使用--dstNode指定它。
如果要将TensorFlow图转换为Onnx图，也可以使用tf2onnx。
您可以使用以下命令转换模型：
python -m tf2onnx.convert --input /frozen_models/model.pb --inputs input --outputs outputNodeName --outputoutput tf_mobilenet.onnx
关于python - 将卡住的图从.pb转换为.onnx，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/57427149/