python - 将卡住图从 .pb 转换为 .onnx

Question

我有一个卡住为 .pb 的 TensorFlow 图，我想将其转换为 .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.

Answer 1

Onnx 格式要求在模型中指定输出节点。快速浏览一下，发现 mmconvert 希望使用 --dstNode 来指定。

如果您要将 TensorFlow 图转换为 Onnx 图，您还可以使用 tf2onnx .

您可以使用以下命令转换模型:

python -m tf2onnx.convert --input /frozen_models/model.pb --inputs input --outputs outputNodeName --outputoutput tf_mobilenet.onnx