PyTorch의 ImageFolder 데이터 로더 활용법
오늘은 ImageFolder를 이용한 이미지 데이터 로딩 방법에 대해 알아보겠습니다. 기본적인 사용 예는 다음과 같습니다:
from torchvision.datasets import ImageFolder
from torchvision import transforms
data_transform = transforms.Compose([
transforms.Resize((224, 224)),
transforms.ToTensor(),
])
dataset = ImageFolder(root='./data', transform=dat ...
7월 3일 21:41에 게시됨