AutoContrast 层

[源代码]

AutoContrast

keras.layers.AutoContrast(value_range=(0, 255), **kwargs)

对图像执行自动对比度操作。

自动对比度会将图像的值拉伸到整个可用的 value_range 范围内。这使得像素之间的差异更加明显。例如,如果一个图像的值范围是 [0, 255],但图像中只有值 [0, 1],则自动对比度会将 1 的值变为 255

此层在训练和推理时都有效。

参数

  • value_range: 输入图像的值范围。表示为包含两个数字的元组,格式为 (low, high)。根据您的预处理管线设置,这通常是 (0, 1)(0, 255)。默认为 (0, 255)