torchvision 입문과 텐서 변환
ndarray → tensor → 정규화 → 모델 입력 형식까지의 변환 흐름을 단계별로 만듭니다.
Web에서 개념과 코드 학습운영체제 권한이 필요한 실행은 Local에서 이어집니다.
01
1단계. 라이브러리 임포트
학습에 필요한 모든 라이브러리를 한 번에 가져옵니다.
- torchvision.transforms로 표준 전처리를 한 줄에 끝낼 수 있습니다.
- 채널 순서 (H, W, C) ↔ (C, H, W) 변환과 dtype 캐스팅을 직접 처리할 수 있습니다.
- 사전학습 가중치 객체에서 권장 transforms와 카테고리 라벨을 꺼낼 수 있습니다.
직접 해보기torch가 CPU 모드에서 동작하고 있는지 확인하세요.
import torch
import torchvision
from torchvision import transforms
import numpy as np
import matplotlib.pyplot as plt
from sklearn.datasets import load_sample_image
torch.__version__, torchvision.__version__