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