STUDY/인공지능
tensor 이미지 저장하는 법
채소보끔
2023. 4. 10. 15:35
728x90
tensor 이미지를 저장하는 방법은 아래 코드와 같다.
from torchvision.utils import save_iamge
save_image(img, '저장하고자하는 경로/이미지 명')
아래 캡쳐본과 같이 실제로 해보면?
아래 캡쳐 사진과 같이 잘 저장됨.
728x90