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 的示例

transforms 的图示

transforms 的图示

forward(img)

参数:

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

返回:

裁剪后的图像。

返回类型:

PIL Image 或 Tensor