使用 Keras 进行数据增强可生成几乎白色的图像

Adr*_*dod 2 python keras data-augmentation

我正在尝试使用 keras ImageDataGenerator 进行数据增强。我正在从数据框中提取图像,该数据框中包含一列中图像的路径和另一列中的标签。目前,我只是尝试水平翻转图像。但是当我绘制图像时,图像看起来亮度被推到了最大值。我想知道这里发生了什么...有什么想法吗?

datagen = ImageDataGenerator(horizontal_flip=True)


# Configure batch size and retrieve one batch of images
for X_batch, y_batch in datagen.flow_from_dataframe(dataframe=data,
    x_col="File name",
    y_col="Driving direction",
    directory = "Self-Driving-Car/Training Data/",
    target_size = (480, 640),
    class_mode = "other"):
    # Show 9 images
    for i in range(0, 9):
        plt.subplot(330 + 1 + i)
        plt.imshow(X_batch[i])
    plt.show()
    break

Run Code Online (Sandbox Code Playgroud)

这是一张原始图像

这是增强图像

Adr*_*dod 5

好吧,这只是一个绘图问题,这解决了它:

    plt.imshow(X_batch[i]/255)
Run Code Online (Sandbox Code Playgroud)