반응형 딥러닝 #파이토치 #Pytorch1 Pytorch - Tensor To Image torch version : 1.7.0 import torchvision torchvision.utils.save_image( [이미지 텐서 변수], [저장 경로]) Ex) torchvision.utils.save_image( img_tensor, os.path.join(path_, 'test.jpg') ) 2021. 8. 28. 이전 1 다음 반응형