TenCrop¶
- class torchvision.transforms.v2.TenCrop(size: Union[int, Sequence[int]], vertical_flip: bool = False)[source]¶
將影像或影片裁剪成四個角落和中心裁剪,加上這些的翻轉版本(預設使用水平翻轉)。
如果輸入是
torch.Tensor
或Image
或Video
,它可以有任意數量的開頭批次維度。 例如,影像可以具有[..., C, H, W]
形狀。請參閱
FiveCrop
以取得範例。注意
此轉換會回傳一個影像的 tuple,因此您的 Dataset 回傳的輸入和目標數量可能不符。 請參閱以下範例,了解如何處理這種情況。
- 參數: