티스토리 뷰
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 |
댓글
250x250
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- support set
- 파이썬
- 프롬프트
- 구글드라이브다운
- 데이터셋다운로드
- clip
- 퓨샷러닝
- 구글드라이브연동
- Unsupervised learning
- 도커
- Prompt
- CNN
- few-shot learning
- 도커 컨테이너
- 구글드라이브서버다운
- vscode 종료
- python
- 구글드라이브서버연동
- 서버구글드라이브연동
- docker
- style transfer
- prompt learning
- NLP
- 서버에다운
- vscode 저장
- 딥러닝
- vscode 자동 저장
- stylegan
- cs231n
- vscode 업데이트
일 | 월 | 화 | 수 | 목 | 금 | 토 |
---|---|---|---|---|---|---|
1 | 2 | 3 | 4 | 5 | 6 | 7 |
8 | 9 | 10 | 11 | 12 | 13 | 14 |
15 | 16 | 17 | 18 | 19 | 20 | 21 |
22 | 23 | 24 | 25 | 26 | 27 | 28 |
29 | 30 |
글 보관함