TenCrop

class torchvision.transforms.v2.TenCrop(size: Union[int, Sequence[int]], vertical_flip: bool = False)

将图像或视频裁剪为四个角落和中心裁剪部分，以及这些部分的翻转版本（默认使用水平翻转）。

如果输入是torch.Tensor或Image或视频，它可以有任意数量的前导批次维度。例如，图像可以有[..., C, H, W]形状。

参见 FiveCrop 的示例。

注意

此转换返回一组图像的元组，并且您的数据集返回的输入和目标数量可能不匹配。请参阅以下示例以了解如何处理这种情况。

Parameters:

• size (序列或整数) – 期望的裁剪输出大小。如果 size 是一个整数而不是像 (h, w) 这样的序列，则会生成一个正方形裁剪 (size, size)。如果提供一个长度为 1 的序列，它将被解释为 (size[0], size[0])。

• vertical_flip (bool, optional) – 使用垂直翻转代替水平翻转