均衡化层
Equalization 类
keras_cv.layers.Equalization(value_range, bins=256, **kwargs)
均衡化在通道级别执行直方图均衡。
参数
- value_range: 一个元组或一个包含两个元素的列表。第一个值表示传递到图像中的值的较低界限，第二个值表示较高的界限。传递到该层的图像应具有
value_range范围内的值。 - bins: 整数，表示直方图均衡中使用的箱子数量。应在 [0, 256] 范围内。
示例
equalize = Equalization()
(images, labels), _ = keras.datasets.cifar10.load_data()
# Note that images are an int8 Tensor with values in the range [0, 255]
images = equalize(images)
调用参数
- images: 像素张量，范围为 [0, 255]，RGB 格式。可以是浮点数或整数类型。应为 NHWC 格式。