在预测Tensorflow中重新训练的Inception-V3时无法找到输入错误

Question

我目前正试图在TensorFlow中对重新训练的Inception-V3 模型进行预测。 当我尝试使用

对图像进行推理时

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

我收到了一个错误 E tensorflow/examples/label_image/main.cc:303] Running model failed: Not found: FeedInputs: unable to find feed output Mul

我使用转移学习来微调在Imagenet数据集上训练的Inception，训练我自己的1000多个课程。培训＆amp;评估过程还可以。我使用tf.train.write_graph()导出图表并使用https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/tools/freeze_graph.py

冻结它

有没有人遇到过这个问题？

Answer 1

似乎在您使用的图表中，Feed或“input_layer”节点已重命名，不再称为“Mul”。您需要找到应将输入注入已保存图形的节点的名称，并通过--input_layer标志传递节点名称。

查找节点名称的最简单方法就是确保在第一次构建图形时明确将其设置为您知道的内容。