本文将带您入门 PyTorch 的图像处理,包括基本的图像读取、转换和可视化等操作。

1. 图像读取

使用 torchvision 库可以方便地读取图像。

import torchvision.transforms as transforms
from torchvision.io import read_image

img = read_image("path/to/image.jpg")

2. 图像转换

PyTorch 提供了一系列图像转换的函数,例如 ToTensor()Normalize() 等。

transform = transforms.Compose([
    transforms.ToTensor(),
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
])

img = transform(img)

3. 图像可视化

使用 matplotlib 库可以将图像可视化。

import matplotlib.pyplot as plt

plt.imshow(img.permute(1, 2, 0))
plt.show()

4. 附加资源

更多关于 PyTorch 图像处理的教程,请参考 PyTorch 图像处理教程

5. 示例图片

下面是一张示例图片:

puppy