所以,我正试图通过他们的github(“https://github.com/tensorflow/tensorflow/blob/master/tensorflow/examples/image_retraining/retrain.py”)的retrain.py代码重新训练图像分类器模型Inception 3 (代码非常冗长,所以我想我不应该在这里发布)
但是,我似乎无法找出存储我想要使用的图像或我应该存储它们的目录的正确方法。我正在使用Windows 7和Python 3.5
目前,我将这些图像放在一个名为Images
的文件夹中,该文件夹在子文件夹中进一步细分,这些子文件夹是根据我想要的标签命名的。
举个例子,它就像Images -> 'Cats'
-> 'Dogs'
-> 'Fish'
其中,Images是主文件夹,子文件夹是动物的类型。
当我运行mnist tutorial.py时,Images
文件夹位于{Mnner教程自动下载Mnist数字数据集的位置C:\tmp\
。所以我想我应该把我的图像存储在同一个位置。
当我从上面的github链接运行retrain.py
时:
我收到以下错误
Image directory "" not found.
File "C:\Users\Amit\AppData\Local\Programs\Python\Python35\Lib\site-packages\tensorflow\models\image\imagenet\tf_retrain.py", line 668, in main
class_count = len(image_lists.keys())
AttributeError: 'NoneType' object has no attribute 'keys'
我查看了代码,并试图找到Image_dir
的默认值,
def create_image_lists
是函数make world
的参数(这似乎是脚本查找图像的位置,但是可以找不到,或者我可能没有正确理解代码。
谢谢。
答案 0 :(得分:1)
您是否可以在此处更改路径的默认文件夹路径(C:\ tmp \ Images)? https://github.com/tensorflow/tensorflow/blob/master/tensorflow/examples/image_retraining/retrain.py#L903