如何使用Keras进行离线图像增强？

Question

我希望在创建模型之前，为数据集中的不同图像类别执行离线图像增强，并将图像保存到其中一个文件夹中。

使用Keras的ImageDataGenerator - flow_from_directory()函数，它具有save_to_dir参数并将其设置为我的目标文件夹。

datagen = ImageDataGenerator(
        rescale=1./255,
        shear_range=0.2,
        zoom_range=0.2,
        horizontal_flip=True)

datagen_set = datagen.flow_from_directory(
        'C:/Users/...',
        target_size=(64, 64),
        batch_size=32,
        classes = ['class_A'],
        save_to_dir = 'C:/Users/.../AugImages',
        save_prefix = 'class_A',
        save_format = 'jpg')

我是否必须创建自己的模型，并使用其fit_generator/fit方法，或者还有其他方法可以在创建模型之前执行离线图像增强？

- k92

Answer 1

如果您想保存增强图像，您需要定义一个模型并使用fit/fit_generator。请注意，datagen_set是一个迭代器，因此您可以使用next方法从迭代器中获取值。

for i in range(no_iter):
    image, label = next(datagen_set)

以上代码将保存batch_size*no_iter张图片到save_to_dir中。

- Lokesh Kumar