python - tensorflow 属性错误: 'NoneType' object has no attribute '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

我从下载了 retrain.py https://github.com/tensorflow/tensorflow/blob/master/tensorflow/examples/image_retraining/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 示例: example (所以不允许我发布图片。)

如果文件存储在 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