通道混洗层

[源代码]

ChannelShuffle 类

keras_cv.layers.ChannelShuffle(groups=3, seed=None, **kwargs)

对输入图像的通道进行混洗。

输入形状

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

输出形状

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

参数

• groups: 用于将输入通道划分的组数，默认为 3。
• seed: 整数。用于创建随机种子。

示例

(images, labels), _ = keras.datasets.cifar10.load_data()
channel_shuffle = ChannelShuffle(groups=3)
augmented_images = channel_shuffle(images)