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})