티스토리 뷰

Skills/PyTorch

[PyTorch] vutils.save_image 텐서 형태 이미지 저장하는 법

미남잉 2023. 7. 26. 13:27
728x90

 

torchvision의 vutils.save_image을 통해 tensor 형태의 이미지를 저장할 수 있다.

 

import torch
import torchvision.utils as vutils

vutils.save_image(out_img,
                  output_path,
                  nrow=1,
                  normalize=True)

 

tensor: 저장할 이미지 텐서(shape은 batch_size, channels, height, width) 이어야 함

filename: 저장할 파일의 경로와 이름

nrow: 저장할 이미지를 몇 줄에 보여줄 것인지

normalize: 이미지의 값을 [0, 1]로 졍규화할 것인지 결정하는 인자. True

728x90
저작자표시 비영리 변경금지

'Skills > PyTorch' 카테고리의 다른 글

[PyTorch] ADJUST_CONTRAST 이미지 대비 (torchvision.transforms.functional)  (14) 2023.07.27
[PyTorch] Optimizer(옵티마이저) 최적화의 단계와 역할  (2) 2023.07.26
[PyTorch] STATE_DICT, CHECKPOINT 에 대해  (0) 2023.07.25
[PyTorch] nn.Parameter() / nn.Variable()  (0) 2023.07.10
[PyTorch] learning rate scheduler - StepLR에 대해  (0) 2023.06.22
댓글