반응형 tensor1 [Python] tensor 를 이미지로 저장하기 방법 1. torchvision.utils 사용 from torchvision.utils import save_image save_image(tensor_data, 'image_path/image_name.png') 방법 2. PIL 이미지로 변경 후 저장 transforms 를 이용하여 Image 로 변경 import torchvision.transforms as T from PIL import Image transform = T.ToPILImage() img = transform(tensor_data) img.save("image_path.png") 방법 3. tensor -> numpy -> Image import numpy as np from PIL import Image np_arr = np.a.. 2022. 6. 20. 이전 1 다음 728x90 반응형
