中心裁剪

class torchvision.transforms.v2.CenterCrop(size: Union[int, Sequence[int]])

在中心裁剪输入。

如果输入是torch.Tensor或TVTensor（例如Image， 视频，BoundingBoxes等）， 它可以有任意数量的前导批次维度。例如， 图像可以有[..., C, H, W]形状。边界框可以有[..., 4]形状。

如果图像的任意一边尺寸小于输出尺寸，图像将用0填充，然后进行中心裁剪。

Parameters:

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

使用 CenterCrop 的示例：

变换的图示

Illustration of transforms