RandomHue 层

[源代码]

RandomHue 类

keras.layers.RandomHue(
    factor, value_range=(0, 255), data_format=None, seed=None, **kwargs
)

随机调整给定图像的色相。

此层将随机增加/减少输入 RGB 图像的色相。

通过将图像转换为 HSV 并旋转色相通道（H）一个增量来调整图像的色相。然后将图像转换回 RGB。

注意：该层可以在 tf.data 或 grain 管道中使用（无论您使用的是哪个后端），且是安全的。

参数

• factor：一个浮点数或两个浮点数的元组。factor 控制图像色相受影响的程度。factor=0.0 使此层执行无操作，而 1.0 执行可用最激烈的对比度调整。如果使用元组，将为每个增强的图像在两个值之间采样一个 factor。如果使用单个浮点数，将从 0.0 和传递的浮点数之间采样一个值。为了确保值始终相同，请传递一个包含两个相同浮点数的元组：(0.5, 0.5)。
• value_range: 输入图像将具有的值的范围。表示为由两个数字组成的元组，写作 [low, high]。这通常是 [0, 1] 或 [0, 255]，具体取决于你的预处理管道的设置方式。
• seed：整数。用于创建随机种子。

示例

(images, labels), _ = keras.datasets.cifar10.load_data()
random_hue = keras.layers.RandomHue(factor=0.5, value_range=[0, 1])
images = keras.ops.cast(images, "float32")
augmented_images_batch = random_hue(images[:8])