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