将keras ImageDataGenerator与多个预处理功能一起使用

时间:2018-09-22 15:13:18

标签: python keras deep-learning data-augmentation

假设我想用Keras训练图像数据库,并想使用Keras ImageDataGenerator自动生成新图像,事实是某些功能在经典设置下不可用(翻转,移位等。)

是否可以不仅添加一个功能,还可以添加一系列功能作为“预处理功能”?

map

我也尝试在调用datagen之前应用我的函数,但是我有太多函数以至于出现MemoryErrors

1 个答案:

答案 0 :(得分:0)

根据评论更新答案

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

也许有更快的班轮可以做到这一点,但它足够干净,可以使工作正常。