CutMix¶

class torchvision.transforms.v2.CutMix(*, alpha: float = 1.0, num_classes: Optional[int] = None, labels_getter='default')[源代码]¶

将 CutMix 应用于提供的图像和标签批次。

注意

此变换旨在用于批次样本，而不是单个图像。有关详细用法示例，请参阅如何使用 CutMix 和 MixUp。样本配对是确定的，并通过匹配批次中连续的样本来完成，因此批次需要进行洗牌（这是一个实现细节，而不是保证的约定。）

在输入中，标签应为形状为 (batch_size,) 的张量。它们将转换为形状为 (batch_size, num_classes) 的张量。

参数:

alpha (float, 可选) – 用于混淆的 Beta 分布的超参数。默认值为 1。
num_classes (int, 可选) – 批次中类的数量。用于独热编码。仅当标签已经独热编码时才能为 None。
labels_getter (可调用对象 或 "default", 可选) – 指示如何在输入中识别标签。默认情况下，如果它是张量，则将选择第二个参数作为标签。这涵盖了将此变换称为 CutMix()(imgs_batch, labels_batch) 的最常见情况。它也可以是可调用对象，接受与变换相同的输入，并返回标签。

使用 CutMix 的示例

文档