图像数据增强
导入相关的包
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')