数据增强

from keras.processing.image import ImageDataGenerator,array_to_img,img_to_array,load_img
import numpy as np

datagen =ImageDataGenerator(
rotation_range = 40,#随机旋转度数，（0，180）
width_shift_range=0.2,#随机水平平移，（0，1）
height_shift_range=0.2,#随机高度平移，（0，1）
rescale =  1/255,#数据归一化，图像都是0-255
shear_range = 0.2,#随机剪切，图像扭曲程度，30能看到扭曲
zoom_range = 0.2,#随机放大
horizontal_flip = True,#随机镜像反转
#还有垂直反转，使用很少
fill_mode = 'nearest',#填充方式，指定位移、旋转后空余像素区域的填充方式
)

#载入图片
img = load_img('')
x = img_to_array(img)
x = np.expand_dims(x,0)#可以reshape

i=0
for batch in datagen.flow(x,batch_size = 1,save_to_dir='保存路径',
    save_prefix='给这个图片数据增强后的所有土拍你提供一前缀',save_foramt='jepg保存格式'）:
    i +=1
    if i == 20 :break#每个图片随机生成20个