将 Pytorch 4D 张量另存为图像

马特帕茨

我有一个 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] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章