Keras fit_generator - ImageDataGenerator 行为

Question

我使用fit function 有一段时间了，但现在我需要对我拥有的数据集进行一些扩充，所以我必须使用fit_generator 。 关于fit_generator有一些我不明白的事情。

为什么流 function 中有一个batch_size选项？ 当我将它与steps_per_epoch参数一起使用时，它的行为如何？ 有没有办法知道使用 imagedatagenerator function 生成了多少图像？

model.fit_generator(datagen.flow(x, y,batch_size=32), steps_per_epoch=len(x)/32,epochs=50)

Answer 1

数据生成器生成批量数据，这意味着对于图像数据，它会生成形状为(batch_size, image_height, image_width, channels)的 numpy arrays 。 fit_generator接收这些批次（显然，对它们进行训练）。 steps_per_epoch定义了一个 epoch 中的批次数。 每批生成的图像数量是你告诉它的，并且图像的总数是无限的（尽管它最终会循环并开始产生重复）