将 Pytorch 4D 张量另存为图像

Question

我有一个 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')

Answer 1

在 PyTorch 中，此代码段正在运行并保存图像：

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在任何情况下检查图像的形状。