Keras 3 API 文档 / KerasCV / / 增强层 / 随机通道移位层

随机通道移位层

[源代码]

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)