我目前正在尝试在TensorFlow中对 re-trained 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个 class 上训练 . 培训和评估过程还可以 . 我用 tf.train.write_graph()
导出图形并用https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/tools/freeze_graph.py冻结它
有人遇到过这个问题吗?
1 回答
似乎在您使用的图表中,Feed或“input_layer”节点已重命名,不再称为“Mul” . 您需要找到应将输入注入已保存图形的节点的名称,并通过--input_layer标志传递节点名称 .
查找节点名称的最简单方法就是确保在第一次构建图形时将其明确设置为您知道的内容 .