在Keras ImageDataGenerator流方法中保存_到_dir时遇到问题

在Keras ImageDataGenerator流方法中保存_到_dir时遇到问题,keras,Keras,我想保存ImageDataGenerator正在创建的增强图像,以便以后使用。当我执行以下代码时,它运行良好,但是我希望保存的图像没有显示在我试图保存它们的目录中 gen = image.ImageDataGenerator(rotation_range=17, width_shift_range=0.12, height_shift_range=0.12, zoom_range=0.12, horizontal_flip=True, dim_order

我想保存ImageDataGenerator正在创建的增强图像,以便以后使用。当我执行以下代码时,它运行良好,但是我希望保存的图像没有显示在我试图保存它们的目录中

gen = image.ImageDataGenerator(rotation_range=17, width_shift_range=0.12,
                     height_shift_range=0.12, zoom_range=0.12, horizontal_flip=True, dim_ordering='th')

batches = gen.flow_from_directory(path+'train', target_size=(224,224),
        class_mode='categorical', shuffle=False, batch_size=batch_size, save_to_dir=path+'augmented', save_prefix='hi')

我觉得我一定没有正确使用此功能。知道我做错了什么吗?

目录中的gen.flow\u为您提供了一个生成器。图像不是真正生成的。为了获得图像,可以遍历生成器。比如说

i = 0
for batch in gen.flow_from_directory(path+'train', target_size=(224,224),
    class_mode='categorical', shuffle=False, batch_size=batch_size,
    save_to_dir=path+'augmented', save_prefix='hi'):

    i += 1
    if i > 20: # save 20 images
        break  # otherwise the generator would loop indefinitely

这只是一个声明,您必须使用该生成器,例如,
.next()


然后您将在
path+'augmented'

中看到图像,我有分类数据-26类。对于每个类,我是否必须创建一个文件夹来保存该类的增强图像(总共26个文件夹)?这将所有增强图像放在一个文件夹中。有没有办法按类别/标签将它们分开?
batches.next()