keras fit_generator 生成了多少张图片？

Question

我使用 keras 进行图像增强和分割。我想调查生成的图像数量，所以我测试了以下参数设置： (1) 定义生成器时在flow_from_directory中设置batch_size为1：

def myGene(...):
    ...
    image_datagen = ImageDataGenerator(**aug_dict)
    image_generator = image_datagen.flow_from_directory(...,batch_size = 1,..., save_prefix  = 'view',...)
    mask_datagen = ImageDataGenerator(**aug_dict)
    mask_generator = mask_datagen.flow_from_directory(...,batch_size = 1,..., save_prefix  = 'view',...)
    ...

（2）训练的时候，我设置epochs = 1，steps_per_epoch=1：

model.fit_generator(myGene,steps_per_epoch=1,epochs=1,...)

训练完成后，我预计目录“./view”中只有 1 个图像和 1 个掩码，但实际上我在那里找到了 11 对。

这里有什么问题？是我没有设置一些参数，还是我做错了什么？

Answer 1

在fit_generator 中，有一个名为max_queue_size=10 的隐藏参数，这意味着生成器将尝试在当前训练批次之前生成10 个批次的队列。

您只是看到为填充队列而生成的图像。您可以尝试使用max_queue_size=0，但即使您可能会看到额外的图像。