Tensorflow Inception V3無法加載計算圖

Question

我在花卉數據集上重新訓練了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'

非常感謝任何幫助，謝謝。

Answer 1

我使用以下命令而不是bazel來運行它，我發現它更容易。

python /path_to_file/label_image.py /path_to_image/image.jpeg

首先確保在運行retrain.py后創建圖形並且它位於正確的位置。 （默認在/ tmp /中）。 如果您需要，可以在此處獲取腳本，並根據需要更改文件位置。

注意：如果圖形文件是不是創造了你可能要檢查這個關於運行retrain.py