本文将带您入门 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. 示例图片
下面是一张示例图片: