TenCrop

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

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

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

有关示例，请参阅 FiveCrop。

注意

此变换返回一个图像元组，并且您的 Dataset 返回的输入和目标数量可能不匹配。有关如何处理此问题，请参阅下面的示例。

参数:

• size (sequence 或 int) – 裁剪的期望输出大小。如果 size 是一个整数而不是序列（如 (h, w)），则进行方形裁剪（size, size）。如果提供长度为 1 的序列，则将其解释为 (size[0], size[0])。

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

transform(inpt: Any, params: dict[str, Any]) → Any

用于覆盖自定义变换的方法。

请参阅 如何编写自己的 v2 变换