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

RandomContrast 层

[源代码]

RandomContrast

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

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

这个层将通过一个随机因子随机调整一张或多张图像的对比度。在训练期间,对比度会为每张图像的每个通道独立调整。

对于每个通道,此层会计算该通道图像像素的平均值,然后将每个像素的每个分量 x 调整为 (x - 平均值) * 对比度因子 + 平均值

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

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

输入形状

3D(非批处理)或 4D(批处理)张量,形状为:(..., 高度, 宽度, 通道数),采用 "channels_last" 格式。

输出形状

3D(非批处理)或 4D(批处理)张量,形状为:(..., 高度, 宽度, 通道数),采用 "channels_last" 格式。

参数

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