Pytorch tensor 이미지 저장, 불러오기

반응형

• torchvision.save_image 사용해서 이미지 저장하기
  • torchvision의 save_image는 [0,1] 범위의 값을 가지는 float tensor를 입력받아 이미지를 저장하는 함수. Input range에 주의할 것.

from torchvision.utils import save_image
	save_image(tensor_data, 'image_path/image_name.png')

• PIL 라이브러리 사용해서 이미지 불러오기

import PIL
import torchvision.transforms as transforms

img = PIL.Image.open('img.png')

tf. transforms.ToTensor()
img_t = tf(img)

print(type(timg_t), img_t.size())