我正在尝试使用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