datagen=ImageDataGenerator()
generator=datagen.flow_from_directory(directory)
如何从生成器获取创建的数据和标记numpy数组。
我已尝试generator.__getitem__
但无法正常工作
答案 0 :(得分:1)
dataset_as_tuple = next(generator)
将为您提供(x,y)的元组,其中x是输入数组(通常是图像),而y是标签数组。通过使用next()
,您将获得batch_size
个样本。设置
generator.batch_size = generator.samples
如果要将样本数设为生成器中的样本数。
答案 1 :(得分:0)
在研究了很多之后,我认为除非您更改源代码,否则无法从ImageDataGenerator中检索numpy数组