RandomHue 层

[源代码]

RandomHue

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

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

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

通过将图像转换为 HSV,并按 delta 旋转色调通道 (H) 来调整图像色调。然后将图像转换回 RGB。

参数

  • 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])