Pytorch中使用transforms.ToTensor和transforms.ToPILImage的numpy.ndarray与Tensor和PILImage的转换举例

最新推荐文章于 2025-06-17 16:20:32 发布

原创最新推荐文章于 2025-06-17 16:20:32 发布 · 9.8k 阅读

50 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #transform

笔记专栏收录该内容

21 篇文章

订阅专栏

本文详细解析了Pytorch中numpy、tensor与PIL图像之间的数据转换过程，包括numpy到tensor的维度变化、数据类型调整及归一化，numpy到PIL的类型需求满足，以及tensor到PIL的维度调整，帮助读者理解并掌握不同数据类型间的转换技巧。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

作为刚接触Pytorch的新人，在刚开始时候处理三种数据tensor，numpy，PIL有点弄晕了，而且中间的转换过程需要注意类型，仅此记录下来，如果能有幸帮到大家那更好了。

进入正题，我们首先看看`numpy to tensor`的情况：

import numpy as np
from torchvision import transforms as T

x = np.ones((128, 228, 1))   # 这里numpy维度是HWC
print(x.shape)   # (128, 228, 1)  
x_tensor = T.ToTensor()(x) 
print(x_tensor.size())   # torch.Size([1, 128, 228])，这里tensor维度是CHW
print(torch.from_numpy(x).size())   # torch.Size([128, 228, 1])注意这里维度不变
x_numpy = x_tensor.numpy()   # 将tensor转换为numpy，但是维度也不改变(1, 128, 228)

所以这种情况比较好理解，就是将HWC变成了CHW，当然这个过程还进行了矩阵运算将其值进行了归一化。

接下来看看`numpy to PIL`的情况：

x = np.ones((128, 228, 1))
print(x.shape)
x_pil = T.ToPILImage()(np.uint8(x))   # 注意这里的np.uint8()，为了使其满足类型需求，在这里是必须的
print(x_pil.size)
x_numpy = np.array(x_pil)
print(x_numpy.shape)

# 输出为：
# (128, 228, 1)
# (228, 128)
# (128, 228)

接下来看看`tensor to PIL`的情况：

x = torch.ones((1, 128, 228))
print(x.shape)
x = x.float()   # 有时候从numpy到tensor后再转化为PIL会报错，这一步就是使其满足数据类型，在这里这一句可以去掉。
x_pil = T.ToPILImage()(x)
print(x_pil.size)
x_tensor = T.ToTensor()(x_pil)
print(x_tensor.size())

# 输出为：
# torch.Size([1, 128, 228])
# (228, 128)
# torch.Size([1, 128, 228])

~~可以看到ToPILImage实际上将CHW变成了WH，也去掉了通道这个维度~~ 。
经过评论区的网友提醒，ToPILImage实际上将CHW变成了CWH，只是因为显示原因没有显示C通道，而原本的size并没有改变，因为恢复回去还是3D的。但是Numpy那里确实是发生了变化，大家可以多多注意下。

可能随着版本的更新，上面这些Case可能都会改变，建议大家自己动手尝试下看看是否与本文中结果一致