RandomChannelShift 層

[來源]

RandomChannelShift 類別

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

隨機平移輸入影像之每個通道的值。

輸入影像的值應介於 [0-255] 或 [0-1] 範圍內。

輸入形狀

具有以下形狀的 3D（未批次處理）或 4D（批次處理）張量：(..., 高度, 寬度, 通道數)，採用 channels_last 格式。

輸出形狀

具有以下形狀的 3D（未批次處理）或 4D（批次處理）張量：(..., 高度, 寬度, 通道數)，採用 channels_last 格式。

參數

• value_range：輸入影像值的範圍。以兩個數字的元組表示，寫成 [最小值, 最大值]。根據預處理流程的設定，這通常是 [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)