本文共 587 字，大约阅读时间需要 1 分钟。
在Pytorch中将PIL Image转换为Tensor是一个常见的操作，尤其是在图像处理任务中。以下是实现这一转换的详细步骤：
from torchvision import transform
首先，导入所需的图像处理库，并定义一个包含多个变换的变换组：
transform = transforms.Compose([ transforms.CenterCrop((512, 512)), # 对图片进行中心裁剪 transforms.ToTensor(),])
接着，打开目标图片文件并确保其颜色模式为RGB：
img = Image.open(img_path).convert('RGB') 然后，应用定义的变换组对图片进行处理：
im = transform(img)
最后，将图片的维度扩展一个维度，以便于后续的批量处理：
im = im.unsqueeze_(dim=0)
此时，图片已经被成功转换为Pytorch Tensor格式，其维度为[B, C, H, W]，其中B表示批量大小，C表示通道数，H和W分别表示高度和宽度。
需要注意的是，在实际应用中，图片的大小和通道数可能会根据具体需求进行调整。例如，如果你的任务需要使用不同的分辨率，可以在CenterCrop中指定不同的裁剪尺寸。
转载地址：http://xqtfk.baihongyu.com/