快捷方式

TenCrop

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

将图像或视频裁剪成四个角和中心裁剪,以及它们的翻转版本(默认使用水平翻转)。

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

请参阅 FiveCrop 以获取示例。

注意

此转换返回一个图像元组,输入和目标的数据集返回的数量可能不匹配。以下是一个关于如何处理此问题的示例。

参数:
  • size (序列int) – 裁剪后的所需输出大小。如果大小是 int 而不是序列,如 (h, w),则裁剪成方形 (size, size)。如果提供长度为 1 的序列,则将其解释为 (size[0], size[0])。

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取面向初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并解决问题

查看资源