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') )