В этой статье мы собираемся обсудить, как читать изображение 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()
Выход: