中心裁剪

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

在中心裁剪输入。

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

如果图像大小沿任何边缘小于输出大小，则用 0 填充图像，然后进行中心裁剪。

参数:

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

使用 CenterCrop 的示例

转换说明

转换说明