如何从keras生成器获取创建的numpy数组?

时间:2018-03-29 02:24:17

标签: keras

datagen=ImageDataGenerator()
generator=datagen.flow_from_directory(directory)

如何从生成器获取创建的数据和标记numpy数组。

我已尝试generator.__getitem__但无法正常工作

2 个答案:

答案 0 :(得分:1)

dataset_as_tuple = next(generator)

将为您提供(x,y)的元组,其中x是输入数组(通常是图像),而y是标签数组。通过使用next(),您将获得batch_size个样本。设置

generator.batch_size  = generator.samples

如果要将样本数设为生成器中的样本数。

答案 1 :(得分:0)

在研究了很多之后,我认为除非您更改源代码,否则无法从ImageDataGenerator中检索numpy数组