Как прочитать изображение JPEG или PNG в PyTorch?

Тензоры в Pytorch Программирование и разработка

В этой статье мы собираемся обсудить, как читать изображение JPEG или PNG с помощью PyTorch в Python.

image_read() method

В PyTorch метод image_read() используется для чтения изображения в качестве входных данных и возврата тензора размера [C, H, W], где C представляет количество каналов, а H, W представляет высоту и ширину соответственно. Приведенный ниже синтаксис используется для чтения изображения JPEG или PNG в PyTorch.

Синтаксис: torchvision.io.read_image(p)

Параметр:

p: это путь к изображению с именем изображения.

Возврат: этот метод возвращает тензор размера [C, H, W].

Изображение ниже используется для демонстрации:

Изображение ниже используется для демонстрации

Пример 1

Следующая программа предназначена для понимания того, как читать изображения в формате 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()

Выход:

ующая программа предназначена для понимания того, как

Пример 2

Следующая программа предназначена для понимания того, как читать изображения 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()

Выход:

Следующая программа предназначена для понимания

Читайте также:  Groovy или Java: подробное сравнение и советы по выбору языка
Оцените статью
bestprogrammer.ru
Добавить комментарий