快捷方式

TenCrop

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

將影像或影片裁剪成四個角落和中心裁剪,加上這些的翻轉版本(預設使用水平翻轉)。

如果輸入是 torch.TensorImageVideo,它可以有任意數量的開頭批次維度。 例如,影像可以具有 [..., C, H, W] 形狀。

請參閱 FiveCrop 以取得範例。

注意

此轉換會回傳一個影像的 tuple,因此您的 Dataset 回傳的輸入和目標數量可能不符。 請參閱以下範例,了解如何處理這種情況。

參數:
  • size (序列int) – 裁切後的目標輸出尺寸。 如果 size 是一個整數而不是像 (h, w) 這樣的序列,則會進行正方形裁切 (size, size)。 如果提供的序列長度為 1,則會被解釋為 (size[0], size[0])。

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

transform(inpt: Any, params: Dict[str, Any]) Any[原始碼]

用於覆寫自定義轉換的方法。

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

文件

取得 PyTorch 的完整開發者文件

檢視文件

教學課程

取得初學者和進階開發人員的深度教學課程

檢視教學課程

資源

尋找開發資源並獲得您的問題解答

檢視資源