tf.keras.preprocessing.image.ImageDataGenerator
keras 内置ImageDataGenerator 可以处理数据扩增的需求。
from keras.preprocessing.image import ImageDataGenerator,array_to_img,img_to_array,load_img
datagen=ImageDataGenerator(
rotation_range=40, # 旋转角度范围
width_shift_range=0.2, # 宽度shift
height_shift_range=0.2,
rescale=1./255, #
shear_range=0.2, #
zoom_range=0.2,
horizontal_flip=True,
fill_mode='nearest')
img=load_img('data/train/cats/cat.0.jpg')
x=img_to_array(img)
x=x.reshape((1,)+x.shape)
i=0
for batch in datagen.flow(x,batch_size=1,
save_to_dir='preview',save_prefix='cat',save_format='jpg'):
i+=1
if i>50:
break