En este artículo, discutiremos cómo invertir los colores de una imagen aleatoriamente con una probabilidad dada en Python usando PyTorch .
Método RandomInvert()
El método RandomInvert() acepta PIL e imagen de tensor como entrada. Con tensor, proporcionamos formas en [3, H, W], donde H y W son la altura y el ancho de la imagen. Este método devuelve una imagen invertida de color aleatorio. La siguiente sintaxis se usa para invertir los colores de una imagen al azar con una probabilidad dada.
Sintaxis: torchvision.transforms.RandomInvert(p)(img)
Parámetro:
- img: Esta es nuestra imagen de entrada para ser invertida.
- p: su probabilidad en el rango de (0,1).
Retorno: este método devuelve una imagen invertida de color aleatorio. si p = 0, devolverá la imagen original y si p = 1, devolverá la imagen invertida en color.
La siguiente imagen se utiliza para la demostración:
Ejemplo:
El siguiente programa es para entender cómo invertir los colores de una imagen al azar con una probabilidad dada.
Python3
# import required libraries
import torch
import torchvision.transforms as T
from PIL import Image
# read image from your computer
img = Image.open('a.png')
# invert the colors with probability=1
transform = T.RandomInvert(p=1)
img = transform(img)
# display result
img.show()
Producción:
Publicación traducida automáticamente
Artículo escrito por mukulsomukesh y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA