Keras 3 API 文档 / 层 API / 预处理层 / 图像增强层 / 随机饱和度层

随机饱和度层

[源代码]

RandomSaturation

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

随机调整给定图像的饱和度。

此层将随机增加/减少输入 RGB 图像的饱和度。

参数

  • factor: 一个包含两个浮点数的元组或一个单独的浮点数。factor 控制图像饱和度受影响的程度。factor=0.5 使此层执行无操作。factor=0.0 使图像完全灰度化。factor=1.0 使图像完全饱和。值应介于 0.01.0 之间。如果使用元组,则会为每个增强的图像在两个值之间采样一个 factor。如果使用单个浮点数,则会在 0.0 和传入的浮点数之间采样一个值。为确保值始终相同,请传入包含两个相同浮点数的元组:(0.5, 0.5)
  • value_range: 输入图像的值范围。表示为写成 [low, high] 的两数字元组。这通常是 [0, 1][0, 255],具体取决于您的预处理管道设置。
  • seed: 整数。用于创建随机种子。

示例

(images, labels), _ = keras.datasets.cifar10.load_data()
images = images.astype("float32")
random_saturation = keras.layers.RandomSaturation(factor=0.2)
augmented_images = random_saturation(images)