datagen=ImageDataGenerator()
generator=datagen.flow_from_directory(directory)
如何从发电机中获取创建的数据并标记numpy阵列。
我尝试过generator.__getitem__
,但无法使它起作用
dataset_as_tuple = next(generator)
将为您提供(x,y(的元组,其中x是输入的数组(通常是图像(,y是标签的数组。通过使用next()
,您可以使用batch_size
样本。设置
generator.batch_size = generator.samples
如果您希望样本数量是发电机中的样本数量。
在研究很多后,我认为除非更改源代码