How to crop an image at random location in PyTorch
Last Updated :
22 Apr, 2022
In this article, we will discuss how to pad an image on all sides in PyTorch.
Torchvision.transforms.RandomCrop method
Cropping is a technique of removal of unwanted outer areas from an image to achieve this we use a method in python that is torchvision.transforms.RandomCrop(). It is used to crop an image at a random location in PyTorch. This method accepts images like PIL Image and Tensor Image. 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.RandomCrop(size)
Parameters:
- size: Desired crop size of the image.
Return: it returns the cropped image of given input size.
Image used for demonstration:
Example 1:
In this example, we are transforming the image with a height of 200 and a width of 400.
Python3
import torch
import torchvision.transforms as transforms
from PIL import Image
image = Image. open ( 'pic.jpg' )
transform = transforms.RandomCrop(( 200 , 400 ))
image_crop = transform(image)
image_crop.show()
|
Output:
Example 2:
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.RandomCrop( 300 )
image_crop = transform(image)
image_crop.show()
|
Output:
Like Article
Suggest improvement
Share your thoughts in the comments
Please Login to comment...