灰度层

[源代码]

Grayscale 类

keras_cv.layers.Grayscale(output_channels=1, **kwargs)

Grayscale 是一个预处理层，用于将 RGB 图像转换为灰度图像。输入图像的值应在 [0, 255] 范围内。

输入形状

3D（非批处理）或 4D（批处理）张量，形状为：(..., height, width, channels)，采用 "channels_last" 格式

输出形状

3D（非批处理）或 4D（批处理）张量，形状为：(..., height, width, channels)，采用 "channels_last" 格式

参数

output_channels。输出图像中存在的颜色通道数。output_channels 可以是 1 或 3。形状为 (..., height, width, 3) 的 RGB 图像在 Grayscale 操作后将具有以下形状：a. (..., height, width, 1) 如果 output_channels = 1 b. (..., height, width, 3) 如果 output_channels = 3。

示例

(images, labels), _ = keras.datasets.cifar10.load_data()
to_grayscale = keras_cv.layers.preprocessing.Grayscale()
augmented_images = to_grayscale(images)