使用具有多个预处理函数的 keras ImageDataGenerator



假设我想用 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, ...]
)

在调用数据生成之前,我也尝试应用我的函数,但我有太多的函数,以至于我有内存错误

根据注释更新答案:

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

也许有更快的一个衬里来实现这一点,但它足够干净,可以让事情正常工作。

最新更新