En este artículo, vamos a discutir cómo leer una imagen JPEG o PNG usando PyTorch en Python .
método image_read()
En PyTorch, el método image_read() se usa para leer una imagen como entrada y devolver un tensor de tamaño [C, H, W], donde C representa una cantidad de canales y H, W representa la altura y el ancho respectivamente. La siguiente sintaxis se usa para leer una imagen JPEG o PNG en PyTorch.
Sintaxis: torchvision.io.read_image(p)
Parámetro:
- p: Esta es la ruta de la imagen con el nombre de la imagen.
Retorno: este método devuelve un tensor de tamaño [C, H, W].
La siguiente imagen se utiliza para la demostración:
Ejemplo 1:
El siguiente programa es para entender cómo leer imágenes jpg.
Python3
# Import required libraries
import torch
import torchvision
from torchvision.io import read_image
import torchvision.transforms as T
# read the jpg image
pic = read_image('img.jpg')
# convert this torch tensor to PIL image
PIL_img = T.ToPILImage()(pic)
# display result
PIL_img.show()
Producción:
Ejemplo 2:
El siguiente programa es para entender cómo leer imágenes png.
Python3
# Import required libraries
import torch
import torchvision
from torchvision.io import read_image
import torchvision.transforms as T
# read the png image
pic = read_image('img.png')
# convert this torch tensor to PIL image
PIL_img = T.ToPILImage()(pic)
# display result
PIL_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