TenCrop¶
- class torchvision.transforms.v2.TenCrop(size: Union[int, Sequence[int]], vertical_flip: bool = False)[source]¶
将图像或视频裁剪成四个角和中心裁剪,再加上这些的翻转版本(默认使用水平翻转)。
如果输入是
torch.Tensor
或Image
或Video
,它可以具有任意数量的前导批次维度。例如,图像可以具有[..., C, H, W]
形状。有关示例,请参见
FiveCrop
。注意
此变换返回图像元组,并且您的数据集返回的输入和目标的数量可能不匹配。请参见下面的示例,了解如何处理这种情况。
- 参数: