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