TenCrop

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

將影像或影片裁剪為四個角和中心裁剪，外加它們的翻轉版本（預設使用水平翻轉）。

如果輸入是 torch.Tensor、Image 或 Video，則可以具有任意數量的前導批次維度。例如，影像的形狀可以是 [..., C, H, W]。

請參閱 FiveCrop 的示例。

注意

此變換返回一個影像元組，並且您的 Dataset 返回的輸入和目標數量可能不匹配。有關如何處理此問題，請參閱下面的示例。

引數:

• size (sequence 或 int) – 裁剪的期望輸出尺寸。如果 size 是一個整數而不是像 (h, w) 這樣的序列，則會進行一個正方形裁剪 (size, size)。如果提供一個長度為 1 的序列，它將被解釋為 (size[0], size[0])。

• vertical_flip (bool, optional) – 使用垂直翻轉代替水平翻轉

transform(inpt: Any, params: dict[str, Any]) → Any

用於覆蓋自定義變換的方法。

請參閱 如何編寫自己的 v2 變換