python - 如何查找 ImageDataGenerator 生成的图像数量

Question

嗨，我想问你一个关于 Keras ImageDataGenerator 的问题。我可以确定将创建多少增强图像吗？或者如何在增强后找到训练图像集的大小。在 Keras 文档中，流程函数描述是:“采用 numpy 数据和标签数组，并生成一批增强/标准化数据。无限循环地产生批次。”但是生成了多少图像？

例如下面的代码生成了多少图像？无限的 ？

from keras.preprocessing.image import ImageDataGenerator, array_to_img, img_to_array, load_img
from matplotlib import pyplot
import numpy as np

datagen = ImageDataGenerator(
        rotation_range=40,
        width_shift_range=0.2,
        height_shift_range=0.2,
        shear_range=0.2,
        zoom_range=0.2,
        horizontal_flip=True,
        fill_mode='nearest')
img=load_img('cat.1.jpg')
x=img_to_array(img)
x = x.reshape((1,) + x.shape) 
print x.shape
h=datagen.flow(x)

Answer 1

您说的是对的，您可以从一张图像中生成无限个数字。这实际上是正常的，仅考虑从初始旋转可以生成的图像数量，它是无限的，因为对于每个旋转角度值，您可以生成不同的图像。为了确认这一点，我将显示来自 keras documentation 的代码

for e in range(epochs):
    print('Epoch', e)
    batches = 0
    for x_batch, y_batch in datagen.flow(x_train, y_train, batch_size=32):
        model.fit(x_batch, y_batch)
        batches += 1
        if batches >= len(x_train) / 32:
            # we need to break the loop by hand because
            # the generator loops indefinitely
            break

请注意，他们说生成器无限循环，这证实生成了无限数量的图像