Keras 2 API 文档 / 层 API / 预处理层 / 图像增强层 / RandomContrast 层

RandomContrast 层

[源代码]

RandomContrast

tf_keras.layers.RandomContrast(factor, seed=None, **kwargs)

一个预处理层,可在训练期间随机调整对比度。

此层将在训练期间以随机因子随机调整图像的对比度。对比度在训练期间会独立地针对每个图像的每个通道进行调整。

对于每个通道,此层都会计算通道中图像像素的均值,然后将每个像素分量 x 调整为 (x - mean) * contrast_factor + mean

输入像素值可以是任何范围(例如 [0., 1.)[0, 255]),可以是整数或浮点数类型。默认情况下,该层将输出浮点数。输出值将被裁剪到 [0, 255] 的范围,这是 RGB 颜色的有效范围。

有关预处理层的概述和完整列表,请参阅预处理 指南

输入形状

3D(非批量)或 4D(批量)张量,形状为:(..., height, width, channels),采用 "channels_last" 格式。

输出形状

3D(非批量)或 4D(批量)张量,形状为:(..., height, width, channels),采用 "channels_last" 格式。

参数

  • factor:一个正浮点数(表示为值的比例),或一个表示下限和上限的长度为 2 的元组。当表示为单个浮点数时,下限 = 上限。对比度因子将从 [1.0 - lower, 1.0 + upper] 之间随机选择。对于通道中的任何像素 x,输出将是 (x - mean) * factor + mean,其中 mean 是通道的平均值。
  • seed:整数。用于创建随机种子。