TensorflowでRe-trained Inception-V3を予測しているときにエラーが発生しました

Question

私は現在、TensorFlowでの再トレーニングされたInception-V3モデルを予測しようとしています。 私は私に訓練するために、Imagenetデータセットに訓練された微調整インセプションに転移学習を使用し、私はエラーに E tensorflow/examples/label_image/main.cc:303] Running model failed: Not found: FeedInputs: unable to find feed output Mul

を取得しています

bazel-bin/tensorflow/examples/label_image/label_image \ --graph=/path/output_graph.pb --labels=/path/output_labels.txt \ --output_layer=final_result \ --image=/path/to/test/image

と画像上の推論を実行しようとしています1000以上のクラスを所有しています。トレーニング&評価プロセスはOKだったグラフをtf.train.write_graph()にエクスポートしてフリーズしました。https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/tools/freeze_graph.py

誰でもこの問題に直面しましたか？

Answer 1

あなたが使用しているグラフでは、フィードまたは「input_layer」ノードの名前が変更され、もはや「Mul」と呼ばれていないようです。保存したグラフに入力を挿入するノードの名前を見つけ、--input_layerフラグでノード名を渡す必要があります。

ノード名を見つける最も簡単な方法は、最初にグラフを作成するときに知っているものに明示的に設定することです。