CenterCrop

class torchvision.transforms.CenterCrop(size)

在中心裁剪給定的影像。如果影像是 torch Tensor，則預期具有 […, H, W] 形狀，其中 … 表示任意數量的領先維度。如果影像大小沿任何邊緣小於輸出大小，則影像會以 0 填補，然後在中心裁剪。

參數:

size (sequence 或 int) – 裁剪後所需的輸出大小。如果 size 是 int 而不是像 (h, w) 這樣的序列，則會進行正方形裁剪 (size, size)。如果提供長度為 1 的序列，它將被解釋為 (size[0], size[0])。

使用 CenterCrop 的範例

轉換的圖示

轉換的圖示

forward(img)

參數:

img (PIL Image 或 Tensor) – 要裁剪的影像。

傳回:

已裁剪的影像。

傳回類型:

PIL Image 或 Tensor