python - 将冻结图从 .pb 转换为 .onnx

Question

我有一个 TensorFlow 图冻结为.pb，我想将其转换为.onnx格式。我目前正在尝试使用 mmconvert （来自微软的MMdnn），显然我做错了（见下面的代码）。是否有任何其他推荐的方法将 a 转换.pb为a .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.

score 4 · Accepted Answer

Onnx 格式需要在模型中指定输出节点。快速浏览表明 mmconvert 期望使用 --dstNode 指定。

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

您将使用以下命令转换模型：

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

1 回答 1