Используйте keras ImageDataGenerator с несколькими функциями предварительной обработки - PullRequest
0 голосов
/ 22 сентября 2018

Допустим, я хотел обучить базу данных изображений с помощью Keras, и я хочу автоматически генерировать новые изображения с помощью Keras ImageDataGenerator, дело в том, что некоторые функции недоступны с классическими настройками (флип, смещение и т. Д.)

Можно ли добавить не одну, а список функций в качестве «функции предварительной обработки»?

datagen = dict(
    horizontal_flip=True, 
    vertical_flip=True,
    width_shift_range = 0.1,
    height_shift_range = 0.1,
    data_format = "channels_first",
    preprocessing_function = [foo1, foo2, ...]
)

Я также пытался применить свои функции перед вызовом datagen, но у меня было так много функций, что у меня были MemoryErrors

1 Ответ

0 голосов
/ 22 сентября 2018

Обновление ответа на основе комментариев:

image_gen = ... # without any preprocessing
def mygen():
  for x in image_gen:
    yield x, foo1(x), foo2(x), ...

Возможно, есть более быстрые лайнеры, чтобы достичь этого, но он достаточно чистый, чтобы все заработало.

...