在Keras ImageDataGenerator流方法中保存_到_dir时遇到问题
我想保存ImageDataGenerator正在创建的增强图像,以便以后使用。当我执行以下代码时,它运行良好,但是我希望保存的图像没有显示在我试图保存它们的目录中在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
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()