将 Pytorch 4D 张量保存为图像答案

【问题标题】：Save Pytorch 4D tensor as image将 Pytorch 4D 张量保存为图像
【发布时间】：2021-02-19 07:33:57
【问题描述】：

我有一个 4-d Pytorch 张量，我想以 .jpg 格式保存到磁盘

我的张量大小如下：

print(image_tensor.size())

>>>torch.Size([1, 3, 400, 711])

我可以在我的 IDE 中将整个张量视为一个图像：

ax1.imshow(im_convert(image_tensor))

由于我能够将整个张量视为一个图像，我假设有一种方法也可以将其保存为这样。但是，当我尝试保存图像时，看起来它只保存了蓝色通道。我想将整个张量保存为单个图像。

img1 = image_tensor[0]
save_image(img1, 'img1.jpg')

【问题讨论】：

图像通常最后有颜色通道。

标签： python image pytorch tensor

【解决方案1】：

在 PyTorch 中，这个 sn-p 正在工作并保存图像：

from torchvision.utils import save_image
import torch
import torchvision

tensor= torch.rand(2, 3, 400, 711) 

img1 = tensor[0]
save_image(img1, 'img1.png')

在保存图像之前，您可以检查img1 的形状，以防万一发生了什么事。

【讨论】：