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 or int) – 所需的裁剪输出大小。如果 size 是一个 int 而不是像 (h, w) 这样的序列，将创建一个正方形裁剪 (size, size)。如果提供长度为 1 的序列，它将被解释为 (size[0], size[0])。

• vertical_flip (bool, 可选的) – 使用垂直翻转而不是水平翻转

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

用于自定义变换的方法，可进行覆盖。

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