快捷方式

TenCrop

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

将图像或视频裁剪成四个角和中心裁剪,再加上这些的翻转版本(默认使用水平翻转)。

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

有关示例,请参见 FiveCrop

注意

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

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

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

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

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

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

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

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

查看教程

资源

查找开发资源并获得问题解答

查看资源