MixUp 層

[原始碼]

MixUp 類別

keras.layers.MixUp(alpha=0.2, data_format=None, seed=None, **kwargs)

MixUp 實作 MixUp 資料擴增技術。

引數

  • alpha:介於 0 和 1 之間的浮點數。控制混合強度。較小的值表示較少的混合,而較大的值允許圖像之間更多的混合。預設值為 0.2,建議用於 ImageNet1k 分類。
  • seed:整數。用於建立隨機種子。

參考文獻

範例

(images, labels), _ = keras.datasets.cifar10.load_data()
images, labels = images[:8], labels[:8]
labels = keras.ops.cast(keras.ops.one_hot(labels.flatten(), 10), "float32")
mix_up = keras.layers.MixUp(alpha=0.2)
output = mix_up({"images": images, "labels": labels})