2

假设我想用 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

4

1 回答 1

0

根据评论更新答案:

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

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

于 2018-09-22T16:03:52.990 回答