Keras 3 API 文件 / KerasCV / 圖層 / 增強圖層 / FourierMix 圖層

FourierMix 圖層

[來源]

FourierMix 類別

keras_cv.layers.FourierMix(alpha=0.5, decay_power=3, seed=None, **kwargs)

FourierMix 实现了 FMix 数据增强技术。

參數

  • alpha:Beta 分布的浮点值。伽马分布的反尺度参数。这控制了从中采样平滑值的分布形状。默认为 0.5,这是论文中推荐的值。
  • decay_power:表示衰减幂的浮点值,默认为 3,如论文中推荐的那样。
  • seed:整数。用于创建随机种子。

參考文獻

範例

(images, labels), _ = keras.datasets.cifar10.load_data()
fourier_mix = keras_cv.layers.preprocessing.FourierMix(0.5)
augmented_images, updated_labels = fourier_mix(
    {'images': images, 'labels': labels}
)
# output == {'images': updated_images, 'labels': updated_labels}