TensorFlow 재교육 label_image.py (개시와 간단한 이미지 분류) 표시 객체에 대한

Question

,

내가 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 '는 존재하지 않는 Tensor를 의미하며,'final_result '연산은 그래프에 존재하지 않습니다. "

참조 링크 : https://github.com/tensorflow/tensorflow/blob/r1.3/tensorflow/examples/image_retraining/label_image.py

Answer 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`