Python PyTorch – Función RandomHorizontalFlip()

En este artículo, discutiremos el método RandomHorizontalFlip() en PyTorch Python .

Método RandomHorizontalFlip()

El método RandomHorizontalFlip() del módulo torchvision.transforms se usa para voltear horizontalmente la imagen dada en un ángulo aleatorio con una probabilidad dada. Este método acepta una PIL y una imagen de tensor como entrada. La imagen del tensor es un tensor PyTorch con forma [C, H, W], donde C representa el número de canales y H, W representa la altura y el ancho respectivamente. Este método devuelve una imagen volteada horizontalmente y una imagen original si la probabilidad P es 1 o 0 respectivamente, si P está en el rango entre 0 y 1, entonces P es la probabilidad de devolver la imagen volteada horizontalmente.

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

Parámetro:

• p: p es la probabilidad de que la imagen se voltee en un ángulo aleatorio.
• img: imagen de entrada que se va a voltear.  

Devoluciones: este método devuelve una imagen volteada aleatoriamente en un ángulo aleatorio.

La siguiente imagen se utiliza para la demostración:

 

Ejemplo 1:

En este ejemplo, volteamos una imagen usando el método RandomHorizontalFlip() cuando la probabilidad es 1.

# 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()

Producción:

 

Ejemplo 2:

En este ejemplo, volteamos una imagen usando el método RandomHorizontalFlip() cuando la probabilidad está en el rango de 0 a 1.

# 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()

Producción: