ChannelShuffle 層

[來源]

ChannelShuffle 類別

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

隨機排列輸入影像的通道。

輸入形狀

形狀為 (..., 高度, 寬度, 通道數) 的 3D（未批次處理）或 4D（已批次處理）張量，採用 "channels_last" 格式

輸出形狀

形狀為 (..., 高度, 寬度, 通道數) 的 3D（未批次處理）或 4D（已批次處理）張量，採用 "channels_last" 格式

參數

• groups：將輸入通道分組的組數，預設為 3。
• seed：整數。用於建立隨機種子。

範例

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