將凍結的圖從.pb轉換為.onnx

Question

我有一個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.

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