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 or int) – 所需的裁剪輸出大小。如果 size 是一個 int 而不是像 (h, w) 這樣的序列，將建立一個正方形裁剪 (size, size)。如果提供長度為 1 的序列，它將被解釋為 (size[0], size[0])。

• vertical_flip (bool, 可選的) – 使用垂直翻轉而不是水平翻轉

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

用於自定義變換的方法，可進行覆蓋。

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