CenterCrop

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

在中心裁剪输入。

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

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

参数:

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

使用 CenterCrop 的示例

旋转边界框上的变换

旋转边界框上的变换

转换图示

转换图示

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

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

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