中心裁剪

class torchvision.transforms.CenterCrop(size)

在中心裁剪给定的图像。如果图像为 torch 张量，则它应该具有 […, H, W] 形状，其中 … 表示任意数量的前导维度。如果图像尺寸沿任何边缘都小于输出尺寸，则图像将用 0 填充，然后进行中心裁剪。

参数：

size (序列 或 int) – 裁剪的所需输出大小。如果大小是一个整数而不是序列，例如 (h, w)，则会进行一个正方形裁剪 (size, size)。如果提供一个长度为 1 的序列，它将被解释为 (size[0], size[0])。

使用 CenterCrop 的示例

变换的说明

变换的说明

forward(img)

参数：

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

返回值：

裁剪后的图像。

返回值类型：

PIL Image 或张量