Enregistrer le tenseur Pytorch 4D en tant qu'image

Question

J'ai un tenseur Pytorch 4-d que je voudrais sauvegarder sur le disque comme un .jpg.

Mon tenseur a la taille suivante :

print(image_tensor.size())

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

Je peux voir le tenseur entier comme une image dans mon IDE :

ax1.imshow(im_convert(image_tensor))

Puisque je suis en mesure de visualiser le tenseur entier sous la forme d'une image, je suppose qu'il existe un moyen de l'enregistrer comme tel. Cependant, lorsque j'essaie d'enregistrer l'image, il semble que seul le canal de couleur bleu soit enregistré. Je voudrais enregistrer le tenseur entier comme une seule image.

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

Answer 1

Dans PyTorch, ce snippet fonctionne et enregistre l'image :

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')

Avant de sauvegarder l'image, pouvez-vous vérifier la forme de l'icône de l'image ? img1 en tout cas, quelque chose s'est passé.