image-processing - 使用 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)