Блог

Я пытаюсь выполнить увеличение данных с помощью 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

  

Хорошо, это была просто проблема построения графика, это решило ее :

     plt.imshow(X_batch[i]/255)
  

В более общем плане имейте в виду, что если у вас нет изображений int с диапазоном 0-255, ImageDataGenerator может масштабировать данные до 0-255 в соответствии с определенными значениями по умолчанию для определенных параметров увеличения, но не для других. brightness_range является одним из примеров масштабирования до 0-255.