torch 读取图片

可以使用torchvision中的函数来读取图片，示例代码如下：

import torch
from torchvision import transforms
from PIL import Image

# 定义图片转换器
transform = transforms.Compose([
    transforms.Resize(224),  # 调整图片大小为224x224
    transforms.CenterCrop(224),  # 中心裁剪为224x224
    transforms.ToTensor(),  # 转换为张量
    transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])  # 标准化
])

# 读取图片
img = Image.open('example.jpg')

# 转换图片
img_tensor = transform(img)

# 打印图片张量的形状
print(img_tensor.shape)

其中，transforms.Compose()函数可以将多个图片转换操作组合在一起，transforms.Resize()函数可以调整图片大小，transforms.CenterCrop()函数可以进行中心裁剪，transforms.ToTensor()函数可以将图片转换为张量，transforms.Normalize()函数可以进行标准化。最后，可以使用Image.open()函数来读取图片