CenterCrop

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

Crop the input at the center.

If the input is a torch.Tensor or a TVTensor (e.g. Image, Video, BoundingBoxes etc.) it can have arbitrary number of leading batch dimensions. For example, the image can have [..., C, H, W] shape. A bounding box can have [..., 4] shape.

If image size is smaller than output size along any edge, image is padded with 0 and then center cropped.

Parameters:

size (sequence or int) – Desired output size of the crop. If size is an int instead of sequence like (h, w), a square crop (size, size) is made. If provided a sequence of length 1, it will be interpreted as (size[0], size[0]).

Examples using CenterCrop:

Illustration of transforms

Illustration of transforms

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

Method to override for custom transforms.

See How to write your own v2 transforms