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 是 int 而不是像 (h, w) 这样的序列，则进行方形裁剪 (size, size)。如果提供长度为 1 的序列，则将其解释为 (size[0], size[0])。

使用 CenterCrop 的示例

变换的图示

变换的图示

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

用于覆盖自定义变换的方法。

参见 如何编写您自己的 v2 变换