더북(TheBook)

예제에서 사용할 이미지 데이터에 대한 전처리 방법을 정의합니다.

코드 5-13 이미지 데이터 전처리 방법 정의

data_path = '../chap05/data/catanddog/train' ------ 이미지 데이터가 위치한 경로 지정

transform = transforms.Compose(
                [
                    transforms.Resize([256, 256]),
                    transforms.RandomResizedCrop(224),
                    transforms.RandomHorizontalFlip(),
                    transforms.ToTensor()
]) ------ ①
train_dataset = torchvision.datasets.ImageFolder(
                data_path,
                transform=transform
) ------ ②
train_loader = torch.utils.data.DataLoader(
               train_dataset,
               batch_size=32,
               num_workers=8,
               shuffle=True
) ------ ③

print(len(train_dataset))
신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.