python中将数据增强

from keras.preprocessing.image import ImageDataGenerator
from keras.preprocessing import image
import matplotlib.pyplot as plt
import numpy as np
import 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()