使用Ubuntu,python3.6 我第一次尝试使用flow_from_directory但是在尝试在我的代码之外使用它并且更改了参数之后它仍然无效。
from keras.preprocessing.image import ImageDataGenerator
categoris = ['cat', 'dog']
def create_train_test_data(WIDTH = 128, HEIGHT = 128, test_size_input = 0.2, numcategories = 1, sizeset = 512):
dlist = {}
datagen = ImageDataGenerator()
for i in range(0,numcategories):
dlist[categoris[i] + '_files'] = datagen.flow_from_directory('data/train_orig/' + numcategoris[i] + '*', target_size=(WIDTH, HEIGHT), batch_size=sizeset, class_mode=None)
我得到的是:
AttributeError: 'ImageDataGenerator' object has no attribute 'flow_from_directory'
和
ImageDataGenerator()
不接受任何参数
这也是我关于堆栈和计算的第一个问题。