CenterCrop

class torchvision.transforms.CenterCrop(size)

Crops the given image at the center. If the image is torch Tensor, it is expected to have […, H, W] shape, where … means an arbitrary number of leading dimensions. 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

forward(img)

Parameters:

img (PIL Image or Tensor) – Image to be cropped.

Returns:

Cropped image.

Return type:

PIL Image or Tensor