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