center_crop¶
-
torchvision.transforms.functional.
center_crop
(img: torch.Tensor, output_size: List[int]) → torch.Tensor[source]¶ 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
img (PIL Image or Tensor) – Image to be cropped.
output_size (sequence or int) – (height, width) of the crop box. If int or sequence with single int, it is used for both directions.
- Returns
Cropped image.
- Return type
PIL Image or Tensor
Examples using
center_crop
: