Tensorflow AttributeError：'NoneType'對象沒有屬性'keys'

Question

我正在嘗試使用Tensorflow創建一個簡單的圖像分類器。 從這里https://medium.com/@linjunghsuan/create-a-simple-image-classifier-using-tensorflow-a7061635984a

我在Windows 10（64位）上使用Anaconda2。使用的軟件包將安裝以下新軟件包：

certifi:        2016.2.28-py35_0
pip:            9.0.3-py35_1
python:         3.5.4-0
setuptools:     36.4.0-py35_1
vc:             14-0
vs2015_runtime: 14.0.25420-0
wheel:          0.29.0-py35_0
wincertstore:   0.2-py35_0

我從https://github.com/tensorflow/tensorflow/blob/master/tensorflow/examples/image_retraining/retrain.py下載了retrain.py

當我嘗試使用訓練數據時

python {$your-working_directory}/retrain.py

我收到這些錯誤

[tensorflow] C:\Users\user>python F:\Tensorflow\retrain.py
ERROR:tensorflow:Image directory '' not found.
Traceback (most recent call last):
  File "F:\Tensorflow\retrain.py", line 1409, in <module>
    tf.app.run(main=main, argv=[sys.argv[0]] + unparsed)
  File "F:\Coursera ML\Anaconda2\envs\tensorflow\lib\site-packages\tensorflow\python\platform\app.py", line 126, in run
    _sys.exit(main(argv))
  File "F:\Tensorflow\retrain.py", line 1061, in main
    class_count = len(image_lists.keys())
AttributeError: 'NoneType' object has no attribute 'keys'

在F：\\ Tensorflow中，有一個文件夾包含數據集圖像和retrain.py

我是一個初學者，這是我第一次使用Tensorflow和深度學習。 我遍歷了StackOverflow和Github，但是找不到任何有效的解決方案。 我認為我在對數據集圖像目錄進行錯誤處理。 任何幫助表示贊賞。 提前致謝。 （另外，如果您需要任何其他信息，請告訴我）

Answer 1

您沒有在命令行中提供所有必需的參數

您鏈接的帖子中的Windows示例： 示例 （因此，我不會發布圖像。）

如果文件存儲在C:\\training_data並且假定您的工作目錄為F:\\Tensorflow則命令為

python F:\\Tensorflow\\retrain.py --image_dir C:\\training_data --how_many_training_steps 500 --model_dir F:\\Tensorflow\\inception --output_graph=F:\\Tensorflow\\retrained_graph.pb --output_labels=F:\\Tensorflow\\retrained_labels.txt

Answer 2

對於發現相同錯誤的MAC用戶

“ Palash Taneja”解決方案可以提供幫助（在Mac上，路徑將有所不同）：

MAC用戶從圖像創建訓練模型的示例命令：

python3 code/retrain.py --image_dir flower_photos --bottleneck_dir=bottleneck --how_many_training_steps 500 --model_dir=model --output_graph=graph/retrained_graph.pb --output_labels=graph/retrained_labels.txt

終端工作目錄的文件和目錄樹：（此處ML為工作目錄）

ML /代碼/ retain.py

ML / flower_photos / {鮮花或各種圖片的各種目錄}

ML /圖

ML /模型

ML /瓶頸