Question

我正在试用来自here的诗人的Tensorflow，它重新训练Ubuntu 14.04 CPU上的最后一个版本

按照指南我运行了这段代码

    python tensorflow/examples/image_retraining/retrain.py \
–bottleneck_dir=tf_files/bottlenecks \
–how_many_training_steps 500 \
–model_dir=tf_files/inception \
–output_graph=tf_files/retrained_graph.pb \
–output_labels=tf_files/retrained_labels.txt \
–image_dir=tf_files/flower_photos

我收到以下错误

Image directory '' not found.
Traceback (most recent call last):
  File "tensorflow/examples/image_retraining/retrain.py", line 774, in <module>
    tf.app.run()
  File "/usr/local/lib/python2.7/dist-packages/tensorflow/python/platform/app.py", line 30, in run
    sys.exit(main(sys.argv))
  File "tensorflow/examples/image_retraining/retrain.py", line 671, in main
    class_count = len(image_lists.keys())
AttributeError: 'NoneType' object has no attribute 'keys

知道这个问题与我的图像存储位置有关，我将它们放在主文件夹中，该文件夹正确应用于最后一个命令image_dir=tf_files/flower_photos

所以现在我的问题是，即使图像目录正确，我为什么会收到此错误？

Answer 1

如果你看第774行，那里发生了错误，它是关于摘要目录，而不是image_dir。

您需要在此处设置摘要目录： https://github.com/tensorflow/tensorflow/blob/master/tensorflow/examples/image_retraining/retrain.py#L928

Answer 2

retrain.py现在位于以下位置：

github.com/tensorflow/hub/raw/master/examples/image_retraining/retrain.py

代替

tensorflow/examples/image_retraining

您可以使用此：

curl -LO https://github.com/tensorflow/tensorflow/raw/master/tensorflow/examples/label_image/label_image.py python label_image.py \
--graph=/tmp/output_graph.pb --labels=/tmp/output_labels.txt \

--input_layer=Placeholder \

--output_layer=final_result \

--image=$HOME/flower_photos/daisy/21652746_cc379e0eea_m.jpg

参考： https://www.tensorflow.org/hub/tutorials/image_retraining

Answer 3

您的代码块缺少双连字符。所以，试试这个：

python tensorflow/examples/image_retraining/retrain.py \
–-bottleneck_dir=tf_files/bottlenecks \
–-how_many_training_steps 500 \
–-model_dir=tf_files/inception \
–-output_graph=tf_files/retrained_graph.pb \
–-output_labels=tf_files/retrained_labels.txt \
–-image_dir=tf_files/flower_photos

Answer 4

请使用绝对路径解决此问题。例如/home/pi/Desktop/hub/retrained_data/dataset

Answer 5

https://colab.research.google.com/drive/1hFte7dNUZGVukuR0OY02t0atJgMzn6oc#scrollTo=Bl-yFZqKeezp&forceEdit=true&offline=true&sandboxMode=true

第0步

将retrain.py替换为从以下链接下载文件并将其上传到Google驱动器中的同一文件夹。

https://raw.githubusercontent.com/tensorflow/hub/master/examples/image_retraining/retrain.py

第1步

转到此链接：

https://colab.research.google.com/drive/1hFte7dNUZGVukuR0OY02t0atJgMzn6oc#scrollTo=Bl-yFZqKeezp&forceEdit=true&offline=true&sandboxMode=true

第2步

转到第二章

第3步

通过在脚本部分之前添加以下两行来进行工作：大声笑

这是一个LOL，仅添加了以下两行，并且在脚本部分之前有效，并且有效。

两行：

if not os.path.isdir("../Pokemons-subset"):

     print('Error')

脚本部分：

 %run scripts/retrainn.py   \

   --bottleneck_dir=tf_files/bottlenecks   \

   --how_many_training_steps=4000   \

   --model_dir=tf_files/models/   \

   --summaries_dir=tf_files/training_summaries/mobilenet_0.50_160   \

   --output_graph=tf_files/retrained_graph.pb   \

   --output_labels=tf_files/retrained_labels.txt   \

   --learning_rate=0.01   \

   --architecture=mobilenet_0.50_160   \

   --image_dir=../Pokemons-subset \

   --flip_left_right \

   --random_crop=10 \

   --random_scale=10 \

   --random_brightness=10

Tensorflow retrain.py未找到图像目录错误

5 个答案: