使用 keras.preprocess.image.ImageDataGenerator 时如何定义我自己的自定义图像预处理函数

Question

我发现有一些图像预处理功能没有包含在 keras.preprocess.image.ImageDataGenerator

那么如何将我自己的自定义预处理功能添加到ImageDataGenerator.

Answer 1

实际上 - 您可以指定自己的预处理功能并将其设置在您的ImageGenerator. 例如：

def preprocessor(image):
    # perform augmentations here

进而：

image_generator = ImageDataGenerator(..., preprocessing_function=preprocessor)