STUDY/인공지능

tensor 이미지 저장하는 법

채소보끔 2023. 4. 10. 15:35
728x90

tensor 이미지를 저장하는 방법은 아래 코드와 같다.

from torchvision.utils import save_iamge

save_image(img, '저장하고자하는 경로/이미지 명')

 

아래 캡쳐본과 같이 실제로 해보면?

 

아래 캡쳐 사진과 같이 잘 저장됨.

 

728x90