How to crop an image at center in PyTorch?
Last Updated :
22 Apr, 2022
In this article, we will discuss how to crop an image at the center in PyTorch.
CenterCrop() method
We can crop an image in PyTorch by using the CenterCrop() method. This method accepts images like PIL Image, Tensor Image, and a batch of Tensor images. The tensor image is a PyTorch tensor with [C, H, W] shape, where C represents a number of channels and H, W represents height and width respectively.
Syntax: torchvision.transforms.CenterCrop(size)
Parameters:
- size: Desired crop size of the image.
Return: This method is returns the cropped image of given input size.
Image used for demonstration:
Example 1:
In this example, we are transforming the image at the center. In this, we will get a square image as output.
Python3
import torch
import torchvision.transforms as transforms
from PIL import Image
image = Image. open ( 'a.jpg' )
transform = transforms.CenterCrop( 200 )
image_crop = transform(image)
image_crop.show()
|
Output:
Example 2:
In this example, we are transforming the image with a height of 180 and a width of 300.
Python3
import torch
import torchvision.transforms as transforms
from PIL import Image
image = Image. open ( 'a.jpg' )
transform = transforms.CenterCrop(( 180 , 300 ))
image_crop = transform(image)
image_crop.show()
|
Output:
Like Article
Suggest improvement
Share your thoughts in the comments
Please Login to comment...