[Pytorch] tensor 를 이미지로 저장하기

방법 1. torchvision.utils 사용

from torchvision.utils import save_image

save_image(tensor_data, 'image_path/image_name.png')

 

방법 2. PIL 이미지로 변경 후 저장

transforms 를 이용하여 Image 로 변경

import torchvision.transforms as T
from PIL import Image

transform = T.ToPILImage()
img = transform(tensor_data)

img.save("image_path.png")

방법 3. tensor -> numpy -> Image

import numpy as np
from PIL import Image

np_arr = np.array(tensor, dtype=np.uint8)
img = PIL.Image.fromarray(np_arr)
img.save('path')