RandomHue 圖層

[原始碼]

RandomHue 類別

keras.layers.RandomHue(
    factor, value_range=(0, 255), data_format=None, seed=None, **kwargs
)

隨機調整給定影像的色調。

此圖層將隨機增加/減少輸入 RGB 影像的色調。

影像色調的調整方式是將影像轉換為 HSV,並將色調通道 (H) 旋轉 delta 值。然後影像會轉換回 RGB。

參數

  • factor:單一浮點數或包含兩個浮點數的元組。factor 控制影像色調受影響的程度。factor=0.0 會使此圖層執行空操作,而值 1.0 則會執行可用的最激進對比度調整。如果使用元組,則會為每個擴增的影像在兩個值之間取樣一個 factor。如果使用單一浮點數,則會在 0.0 和傳遞的浮點數之間取樣一個值。為了確保值始終相同,請傳遞包含兩個相同浮點數的元組:(0.5, 0.5)
  • value_range:傳入影像將具有的值範圍。表示為寫成 [low, high] 的雙數字元組。這通常是 [0, 1][0, 255],具體取決於您的預處理管線設定方式。
  • seed:整數。用於建立隨機種子。

範例

(images, labels), _ = keras.datasets.cifar10.load_data()
random_hue = keras.layers.RandomHue(factor=0.5, value_range=[0, 1])
images = keras.ops.cast(images, "float32")
augmented_images_batch = random_hue(images[:8])