python 图片数据增强_python中将数据增强

python中将数据增强

发布时间：2018-01-19 23:29,

浏览次数：493

, 标签：

python

from keras.preprocessing.image import ImageDataGenerator from

keras.preprocessingimport image import matplotlib.pyplot as plt import numpy as

npimport pylab img_generator = ImageDataGenerator( rotation_range = 90,

width_shift_range= 0.2, height_shift_range = 0.2, zoom_range = 0.3 ) # 导入并显示图片

img_path= 'G:\\100000.jpg' img = image.load_img(img_path) print(type(img)) #

plt.imshow(img)# pylab.show() #img.show() # 将图片转为数组 x = image.img_to_array(img)

#扩充一个维度 x = np.expand_dims(x, axis=0) # 生成图片 gen = img_generator.flow(x,

batch_size=2) # 显示生成的图片 plt.figure() for i in range(1): for j in range(1):

x_batch= next(gen) print(type(x_batch),x_batch.shape) idx = (1*i) + j

plt.subplot(1, 1, idx+1) plt.imshow(x_batch[0]/255) pylab.show()