① 데이터 전처리에 사용되는 transforms.Compose 파라미터 중 새롭게 등장하는 것들만 알아보겠습니다.
ⓐ transforms.Resize: 이미지를 주어진 크기로 재조정합니다. 즉, 256×256 크기로 이미지를 조정합니다.
ⓑ transforms.RandomRotation: 5도 이하로 이미지를 회전시킵니다.
ImageFolder를 이용하여 모델 학습에 필요한 데이터셋을 불러옵니다.
코드 6-48 ImageFolder를 이용하여 데이터셋 불러오기
train_path = '../chap06/data/catanddog/train' ------ 훈련 데이터셋이 위치한 경로
test_path = '../chap06/data/catanddog/test' ------ 테스트 데이터셋이 위치한 경로
train_dataset = torchvision.datasets.ImageFolder(
train_path,
transform=train_transforms
) ------ ①
test_dataset = torchvision.datasets.ImageFolder(
test_path,
transform=test_transforms
)
print(len(train_dataset)), print(len(test_dataset))