TensorFlow重新训练label_image.py(使用Inception进行简单图像分类)

时间:2017-10-26 12:41:25

标签: python tensorflow gcloud tensorboard google-cloud-ml-engine

对于标记对象,

如果我使用TensorFlow的label_image.py,如:

    $ python label_image.py --image=1.jpg --graph=output_inference_graph.pb/frozen_inference_graph.pb --labels=test.txt

错误:KeyError:"名称' final_result:0'是指不存在的张量。图表中不存在操作' final_result'"

参考链接:https://github.com/tensorflow/tensorflow/blob/r1.3/tensorflow/examples/image_retraining/label_image.py

1 个答案:

答案 0 :(得分:1)

您缺少在训练期间使用的输出图层参数。 试试

`bazel-bin/tensorflow/examples/image_retraining/label_image \
--graph=/tmp/your_output_graph.pb 
--labels=/tmp/your_output_labels.txt \
--output_layer=final_result:0 \
--image=path_to_your_imahe.jpg`