本文共 587 字，大约阅读时间需要 1 分钟。

PIL Image转Pytorch Tensor

在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中指定不同的裁剪尺寸。