基于鱼皮肤的鱼个体识别（3）- data augmentation

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