CenterCrop

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

从中心裁剪输入。

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

如果图像在任何边缘上的尺寸小于输出尺寸，图像将用 0 填充，然后从中心裁剪。

参数:

size (sequence 或 int) – 期望的裁剪输出尺寸。如果 size 是一个整数而不是像 (h, w) 这样的序列，则进行一个正方形裁剪 (size, size)。如果提供一个长度为 1 的序列，则将其解释为 (size[0], size[0])。

使用 CenterCrop 的示例

转换的图示

转换的图示

transform(inpt: Any, params: Dict[str, Any]) → Any

用于覆盖以实现自定义转换的方法。

请参阅 如何编写自己的 v2 转换