MixUp 层

[源代码]

MixUp

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

MixUp 实现 MixUp 数据增强技术。

注意:该层可以在 tf.datagrain 管道中使用(无论您使用的是哪个后端),且是安全的。

参考文献

参数

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