我在Keras中读过一些关于数据扩充的资料,但对我来说还是有点模糊。在数据扩充步骤中,是否有任何参数来控制从每个输入图像创建的图像的数量?在this example中,我看不到任何控制从每个图像创建的图像数量的参数。在
例如,在下面的代码中,我可以使用一个参数(num_imgs)来控制从每个输入图像创建并存储在名为preview的文件夹中的图像数量;但是在实时数据扩充中,没有任何参数用于此目的。在from keras.preprocessing.image import ImageDataGenerator, array_to_img, img_to_array, load_img
num_imgs = 20
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('data/train/cats/cat.0.jpg') # this is a PIL image
x = img_to_array(img) # this is a Numpy array with shape (3, 150, 150)
x = x.reshape((1,) + x.shape) # this is a Numpy array with shape (1, 3, 150, 150)
# the .flow() command below generates batches of randomly transformed images
# and saves the results to the `preview/` directory
i = 0
for batch in datagen.flow(x, batch_size=1,
save_to_dir='preview', save_prefix='cat', save_format='jpeg'):
i += 1
if i > num_imgs:
break # otherwise the generator would loop indefinitely