Tensor to image / numpy array to image / PIL image save (Save feature as image)

import torchvision

# imgs shape = torch.Size([1, 3, 1056, 736])
torchvision.utils.save_image(imgs,'./test.jpg')

torchvision 내 utils.save_image 함수 적용

 

import cv2

cv2.imwrite('./test.jpg', imgs)

cv2 내 imwrite 함수 적용

 

from PIL import Image
 
# 이미지 열기
im = Image.open('python.png')
 
# 이미지 크기 출력
print(im.size)
 
# 이미지 JPG로 저장
im.save('python.jpg')

PIL 객체 내 save함수 사용