Keras 3 API 文档 / KerasCV / / 预处理层 / 海报化层

海报化层

[源代码]

Posterization

keras_cv.layers.Posterization(value_range, bits, **kwargs)

减少每个颜色通道的比特数。

参考文献

参数

  • value_range:一个包含两个元素的元组或列表。第一个值表示传递图像中值的较低边界,第二个值表示较高的边界。传递给该层的图像应该具有value_range内的值。默认为(0, 255)
  • bits:整数,每个通道保留的比特数。必须是 1-8 之间的值。

示例

(images, labels), _ = keras.datasets.cifar10.load_data()
print(images[0, 0, 0])
# [59 62 63]
# Note that images are Tensors with values in the range [0, 255] and uint8
dtype
posterization = Posterization(bits=4, value_range=[0, 255])
images = posterization(images)
print(images[0, 0, 0])
# [48., 48., 48.]
# NOTE: the layer will output values in tf.float32, regardless of input
    dtype.

调用参数

  • inputs:输入张量,可以有两种格式
    1. 单个 3D (HWC) 图像或 4D (NHWC) 图像批次。
    2. 一个张量字典,其中图像位于"images"键下。