使用 ImageDataGenerator 更改颜色答案

【问题标题】：Change colors with ImageDataGenerator使用 ImageDataGenerator 更改颜色
【发布时间】：2019-12-07 12:07:42
【问题描述】：

我正在使用 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，但它更像是亮度/对比度。

【问题讨论】：

标签： python tensorflow keras

【解决方案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
                )

【讨论】：

【解决方案2】：

在阅读 Medium 上的文章时发现了一些东西。它可能会有所帮助：-

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
    )

来源：-

https://medium.com/@halmubarak/changing-color-space-hsv-lab-while-reading-from-directory-in-keras-c8ca243e2d57

【讨论】：

请在回答中引用文章的相关部分（以防链接失效、文章被删除等）。