Увеличение данных с помощью Keras дает почти белые изображения - PullRequest
Новая
       21

Увеличение данных с помощью Keras дает почти белые изображения

0 голосов
/ 08 апреля 2019

Я пытаюсь сделать увеличение данных с помощью keras ImageDataGenerator. Я вытаскиваю изображения из Dataframe, содержащего пути к изображению в одном столбце и метку в другом. Сейчас я только пытаюсь перевернуть изображение по горизонтали. Но когда я строю изображения, изображения выглядят так, как будто яркость была доведена до максимума. Интересно, что здесь происходит ... Есть мысли? 

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

Here is one original image

Here are the augmented images

1 Ответ

0 голосов
/ 08 апреля 2019

Хорошо, это была просто проблема с заговором, это решило ее: 

    plt.imshow(X_batch[i]/255)
...