python - 使用 ImageDataGenerator 更改颜色

Question

我正在使用 keras ImageDataGenerator 来预处理训练图像，并且需要某种颜色更改功能(随机颜色、色调更改)。

我的生成器代码如下所示:

image_generator = tf.keras.preprocessing.image.ImageDataGenerator(                                 
                              horizontal_flip = True,
                              brightness_range= [0.7, 1.3],
                              rotation_range = 10,
                              zoom_range = [0.8, 1.2],
                              width_shift_range=0.2,
                              height_shift_range=0.2,
                             fill_mode="nearest")

我尝试浏览数据收集器的keras手册，我发现最好的是-channel_shift_range，但它的工作原理更像是亮度/对比度。

Answer 1

也许this可以帮助。您可以定义一个自定义函数以在 ImageDataGenerator 中使用它来修改图像颜色。

例如:

import cv2
import numpy as np
from PIL import Image

def myFunc(image):
    image = np.array(image)
    hsv_image = cv2.cvtColor(image,cv2.COLOR_RGB2HSV)
    return Image.fromarray(hsv_image)

train_datagen = ImageDataGenerator(
                rescale=1. / 255,
                rotation_range=20,
                width_shift_range=0.2,
                height_shift_range=0.2,
                horizontal_flip=True,
                preprocessing_function = myFunc
                )