我在花卉数据集上重新训练了 Tensorflow 的 Inception V3 的最后一层。这是使用以下方法完成的:
bazel-bin/tensorflow/examples/image_retraining/retrain --image_dir ~/flower_photos
训练成功,然后我跑了:
bazel build tensorflow/examples/label_image:label_image
这也运行得很好,接下来我运行:
bazel-bin/tensorflow/examples/label_image/label_image \
--graph=/tmp/output_graph.pb --labels=/tmp/output_labels.txt \
--output_layer=final_result \
--image=$HOME/flower_photos/daisy/21652746_cc379e0eea_m.jpg
当我运行此程序时,我收到一条错误消息
E tensorflow/examples/label_image/main.cc:285] Not found: Failed to load compute graph at '/tmp/output_graph.pb'
非常感谢任何帮助,谢谢。
最佳答案
我使用以下命令而不是 bazel 运行它,我发现它更容易。
python /path_to_file/label_image.py /path_to_image/image.jpeg
首先确保图形是在运行 retrain.py 之后创建的,并且位于正确的位置。 (默认位于/tmp/内)。如果您愿意,您可以获取脚本 here您可以根据需要更改文件位置。
注意:如果未创建图形文件,您可能需要检查 this关于运行 retrain.py
关于python - Tensorflow Inception V3 无法加载计算图,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/42430355/