使用ImageDataGenerator，matplotlib抛出TypeError:图像数据的无效形状（1、256、256、3）

Question

我有15张汽车图像，使用数据增强，我想用它们创建一个数据集。然而，当我使用Keras的ImageDataGenerator并试图绘制生成的图像时，我得到一个错误，即

TypeError: Invalid shape (1, 256, 256, 3) for image data.

我附加的代码，以及，请让我知道我如何可以解决这个问题

datagen = ImageDataGenerator(rescale=1./255, zoom_range=0.1, rotation_range=25, width_shift_range=0.1, height_shift_range=0.1, shear_range=0.1, horizontal_flip=True)

ite = datagen.flow_from_directory("Car Images", batch_size=1)

for i in range(9):

    # define subplot
    plt.subplot(330 + 1 + i)

    # generate batch of images
    batch = ite.next()

    # convert to unsigned integers for viewing
    image = batch[0].astype('uint8')

    # plot raw pixel data
    plt.imshow(image)

# show the figure
plt.show()

错误指向plt.imshow()行

Answer 1

你需要重塑图像，试试这个

datagen = ImageDataGenerator(rescale=1./255, zoom_range=0.1, rotation_range=25, width_shift_range=0.1, height_shift_range=0.1, shear_range=0.1, horizontal_flip=True)

ite = datagen.flow_from_directory("Car Images", batch_size=1)

for i in range(9):

    # define subplot
    plt.subplot(330 + 1 + i)

    # generate batch of images
    batch = ite.next()

    # convert to unsigned integers for viewing
    image = batch[0].astype('uint8')

    image = np.reshape(256,256,3)

    # plot raw pixel data
    plt.imshow(image)

# show the figure
plt.show()