torchvision1 [PyTorch] Torchvision과 utils.data 이수안컴퓨터연구소의 파이토치(PyTorch) 기초 영상을 보고 정리한 내용입니다. ✅ Torchvision Torchvision은 대표적으로 전처리할 때 사용하는 메서드인 transforms를 제공하는 패키지이다. 이외에도 일반적으로 클래스를 따로 만들어 전처리 단계를 진행하며 다양한 전처리 기술을 제공한다. import torch import torchvision import torchvision.transforms as transforms transform = transforms.Compose([transforms.ToTensor(), transforms.Normalize(mean=(0.5,), std=(0.5,))]) 예를 들어 DataLoader의 인자로 들어갈 transform을 미리 정의할 수.. 2022. 2. 23. 이전 1 다음