torchタイプから画像表示する方法

Question

pytorchのデータ型のtochデータを画像表示する方法を教えて頂きたいです。

torch.Size：([1, 3, 64, 64])の4次元のデータ。
1:画像枚数,2:画像チャンネル数(RGB),3-4:各チャネルの各ピクセルのデータ

上記データを画像表示する方法に悩んでおります。
ご教授お願い致します。

Answer 1

画像をどの形式でtensorにしているかによって解決策が変わるかと思いますが、
一般的には以下のようにすれば解決するかと思います

import torch
import matplotlib.pyplot as plt
from torchvision import transforms

tensor # torch.Size：([1, 3, 64, 64])の4次元のデータ。
img = transforms.ToPILImage(mode="RGB")(tensor)
plt.imshow(img)
plt.show()

Comments

[Vincent]

丁寧にありがとうございます！