Python 增加数据集大小的数据扩充

所以在这里，我想对图像数据集应用增强，以增加数据集的大小，但我不断得到错误

代码如下：

**from keras.preprocessing.image import ImageDataGenerator, array_to_img, img_to_array, load_img
import numpy as np
import os
from PIL import Image 

datagen = ImageDataGenerator(
        rotation_range=40,
        width_shift_range=0.2,
        height_shift_range=0.2,
        shear_range=0.2,
        zoom_range=0.2,
        horizontal_flip=True,
        fill_mode='nearest')

image_directory = r'/gdrive/My Drive/HealthySkin/'
SIZE= 128
dataset = []

my_images = os.listdir(image_directory)
for i ,image_name in enumerate(my_images):
       if (image_name.split(".")[1] == 'jpeg'):
        image = io.imread(image_directory + image_name)
        image = Image.fromarray(image,'RGB')
        image = Image.resize((SIZE,SIZE))
        dataset.append(np.array(image))

x = np.array(dataset)


i = 0
for batch in datagen.flow(x, batch_size=16 , save_to_dir='Aug', save_prefix='1', save_format='jpeg'):
    i += 1
    if i > 20:
        break** 

错误显示为“ValueError:（'Input data in

numpyaryarrayiterator

应具有秩4。您传递了一个具有形状'，（0，）”的数组） 最后一个循环 我如何修复错误