MixUp 層
MixUp 類別
keras_cv.layers.MixUp(alpha=0.2, seed=None, **kwargs)
MixUp 实现了 MixUp 資料增強技術。
參數
- alpha：介於 0 到 1 之间的浮點數。伽瑪分佈的反比例參數。這控制了从中採樣平滑值的分配形狀。默认为 0.2，這是訓練 imagenet1k 分類模型時的建議值。
- seed：整數。用於建立隨機種子。
參考
範例
(images, labels), _ = keras.datasets.cifar10.load_data()
images, labels = images[:10], labels[:10]
# Labels must be floating-point and one-hot encoded
labels = tf.cast(tf.one_hot(labels, 10), tf.float32)
mixup = keras_cv.layers.preprocessing.MixUp(10)
augmented_images, updated_labels = mixup(
{'images': images, 'labels': labels}
)
# output == {'images': updated_images, 'labels': updated_labels}