【PIL与Tensor】Pytorch中Tensor与PIL图像格式相互转化
首先,将Tensor复制到CPU并调整维度。然后使用.numpy()函数将Tensor转换为numpy数组,并乘以255以还原为原始图像数据类型。最后使用Image.fromarray()将numpy数组转换为PIL Image。其中,np.array()将PIL Image转换为numpy数组,.permute()调整了数组的维度以适应Pytorch Tensor的格式,并将数据类型转换为float类型。可以看到当前转化是正确的!
原创
2023-05-22 10:02:06 ·
1600 阅读 ·
1 评论