中心裁剪

class torchvision.transforms.CenterCrop(size)

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

Parameters:

size (序列或整数) – 期望的裁剪输出大小。如果 size 是一个整数而不是像 (h, w) 这样的序列，则会生成一个正方形裁剪 (size, size)。如果提供一个长度为 1 的序列，它将被解释为 (size[0], size[0])。

使用 CenterCrop 的示例：

变换的图示

Illustration of transforms

forward(img)

Parameters:

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

Returns:

裁剪后的图像。

Return type:

PIL 图像或张量