CenterCrop

class torchvision.transforms.CenterCrop(size)

在影像中心裁剪給定影像。如果影像是 Torch Tensor，則期望其形狀為 […, H, W]，其中 … 表示任意數量的前導維度。如果影像尺寸沿任何邊小於輸出尺寸，則影像用 0 填充，然後在中心裁剪。

引數:

size (sequence or int) – 裁剪所需的輸出尺寸。如果 size 是 int 而不是 sequence（如 (h, w)），則進行正方形裁剪 (size, size)。如果提供長度為 1 的 sequence，則將其解釋為 (size[0], size[0])。

使用 CenterCrop 的示例

轉換圖示

轉換圖示

forward(img)

引數:

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

返回:

裁剪後的影像。

返回型別:

PIL Image 或 Tensor