不能使用'flow_from_directory',因为它似乎不存在

时间:2018-02-04 12:30:25

标签: ubuntu machine-learning keras python-3.6

使用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()

不接受任何参数

这也是我关于堆栈和计算的第一个问题。

0 个答案:

没有答案