Skip to content
Related Articles

Related Articles

Python PyTorch – RandomHorizontalFlip() Function

View Discussion
Improve Article
Save Article
  • Last Updated : 10 Jun, 2022
View Discussion
Improve Article
Save Article

In this article, we will discuss the RandomHorizontalFlip() Method in PyTorch Python.

RandomHorizontalFlip() method

RandomHorizontalFlip() method of torchvision.transforms module is used to horizontally flip the given image at a random angle with a given probability. This method accepts a PIL and tensor image as input. The tensor image is a PyTorch tensor with shape [C, H, W], where C represents the number of channels and  H, W represents the height and width respectively. This method returns a horizontally flipped image and an original image if the probability P is 1 or 0 respectively, if P is in the range between 0 to 1 then P is the probability to return the horizontally flipped image.

Syntax: torchvision.transforms.RandomHorizontalFlip(p)(img)

Parameter:

  • p: p is the probability of the image being flipped at a random angle.
  • img: input image to be flipped.  

Returns: This method returns a randomly flipped image  at a random angle.

The below image is used for demonstration:

 

Example 1:

In this example, we flip an image using RandomHorizontalFlip() Method when the probability is 1.

Python3




# import required libraries
import torch
import torchvision.transforms as T
from PIL import Image
  
# read input image from computer
img = Image.open('a.png')
  
# define a transform
transform = T.RandomHorizontalFlip(p=1)
  
# apply above defined transform to 
# input image
img = transform(img)
  
# display result
img.show()

Output:

RandomHorizontalFlip() Method in Python PyTorch

 

Example 2:

In this example, we flip an image using RandomHorizontalFlip() Method when the probability is in the range of 0 to 1.

Python3




# import required libraries
import torch
import torchvision.transforms as T
from PIL import Image
  
# read input image from computer
img = Image.open('img.png')
  
# define a transform
transform = T.RandomHorizontalFlip(p=0.5)
  
# apply above defined transform to 
# input image
img = transform(img)
  
# display result
img.show()

Output:

RandomHorizontalFlip() Method in Python PyTorch

 


My Personal Notes arrow_drop_up
Recommended Articles
Page :

Start Your Coding Journey Now!