Keras 图像数据增强

图像数据增强

导入相关的包

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

定义图像生成对象

datagen = ImageDataGenerator(
     rotation_range=40,  #随机旋转度数
    width_shift_range = 0.2,  #随机水平平移
    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('image/content/baixiaona.jpg')
x = img_to_array(img)
print(x.shape)

(1000, 1334, 3)

x = np.expand_dims(x,0)
print(x.shape)

(1, 1000, 1334, 3)

图片生成

#生成20张图片
i = 0
for batch in datagen.flow(x,batch_size=1,save_to_dir='temp',save_format='jpeg'):
    i+=1
    if i==5:
        break
print('finished')