基于Tensorflow深度学习框架下的图像增强(二)

datagen = image.ImageDataGenerator(fill_mode='wrap',zoom_range=[4,4])
gen = image.ImageDataGenerator()
data = gen.flow_from_directory(in_path,batch_size=1,class_mode=None,shuffle = True, target_size = (224,224))
np_data = np.concatenate([data.next() for i in range(data.n)])
datagen.fit(np_data)
gen_data = datagen.flow_from_directory(in_path,batch_size = 1, shuffle = False, 
                                       save_to_dir = out_path+'fill_mode',
                                       save_prefix='gen',target_size=(224,224))
for i in range(3):
    gen_data.next()
print_result(out_path+'fill_mode/*')

datagen = image.ImageDataGenerator(fill_mode='nearest',zoom_range=[4,4])
gen = image.ImageDataGenerator()
data = gen.flow_from_directory(in_path,batch_size=1,class_mode=None,shuffle = True, target_size = (224,224))
np_data = np.concatenate([data.next() for i in range(data.n)])
datagen.fit(np_data)
gen_data = datagen.flow_from_directory(in_path,batch_size = 1, shuffle = False, 
                                       save_to_dir = out_path+'nearest',
                                       save_prefix='gen',target_size=(224,224))
for i in range(3):
    gen_data.next()
print_result(out_path+'nearest/*')