随机通道移位层

[源代码]

RandomChannelShift 类

keras_cv.layers.RandomChannelShift(
    value_range, factor, channels=3, seed=None, **kwargs
)

随机移动输入图像的每个通道的值。

输入图像的值应在 [0-255] 或 [0-1] 范围内。

输入形状

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

输出形状

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

参数

• value_range: 输入图像将具有的值范围。以两个数字元组表示，写为 [low, high]。这通常是 [0, 1] 或 [0, 255]，具体取决于预处理管道的设置方式。
• factor: 标量值，或 [0.0, 1.0] 范围内的两个浮点值的元组/列表。如果 factor 是单个值，则它将被解释为等效于元组 (0.0, factor)。factor 将为每个要增强的图像在它的范围内进行采样。
• channels: 整数，要移动的通道数，默认为 3，对应于 RGB 移位。在某些情况下，可能会有更多或更少的通道。
• seed: 整数。用于创建随机种子。

示例

(images, labels), _ = keras.datasets.cifar10.load_data()
rgb_shift = keras_cv.layers.RandomChannelShift(value_range=(0, 255),
    factor=0.5)
augmented_images = rgb_shift(images)