我有一个 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')
在 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
在任何情况下检查图像的形状。
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句