Tensorflow AttributeError: ‘NoneType’ 对象没有属性 ‘keys’

我正在尝试使用Tensorflow创建一个简单的图像分类器。参考自这里https://medium.com/@linjunghsuan/create-a-simple-image-classifier-using-tensorflow-a7061635984a

我在Windows 10(64位)上使用Anaconda2。使用的软件包如下,这些新软件包将被安装:

certifi:        2016.2.28-py35_0pip:            9.0.3-py35_1python:         3.5.4-0setuptools:     36.4.0-py35_1vc:             14-0vs2015_runtime: 14.0.25420-0wheel:          0.29.0-py35_0wincertstore:   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.pyERROR: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上搜索了很多,但没有找到任何有效的解决方案。我认为我可能在数据集图像的目录设置上犯了错误。任何帮助都将不胜感激。提前感谢您。(如果您需要更多信息,请告诉我)


回答:

您在命令行中没有提供所有必需的参数

您链接的文章中提到的Windows示例:示例 (StackOverflow不允许我发布图片。)

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

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注