728x90
tensor 이미지를 저장하는 방법은 아래 코드와 같다.
from torchvision.utils import save_iamge
save_image(img, '저장하고자하는 경로/이미지 명')
아래 캡쳐본과 같이 실제로 해보면?
아래 캡쳐 사진과 같이 잘 저장됨.
728x90
'STUDY > 인공지능' 카테고리의 다른 글
| torch.unsqueeze , torch.squeeze, torch.cat 설명 (0) | 2023.04.19 |
|---|---|
| Diffusion model (1) (0) | 2023.04.18 |
| EasyDict 이란? 사용법 (0) | 2023.03.29 |
| yaml 파일 다루기, 읽기, key value 값 뽑기 (0) | 2023.03.29 |
| np.array, ToTensor, ToPILImage 공부 (0) | 2023.03.28 |
