더북(TheBook)

테스트 데이터를 불러와 전처리를 합니다.

코드 5-23 테스트 데이터 호출 및 전처리

test_path = '../chap05/data/catanddog/test'

transform = transforms.Compose(
                [
                   transforms.Resize(224),
                   transforms.CenterCrop(224),
                   transforms.ToTensor(),
])
test_dataset = torchvision.datasets.ImageFolder(
    root=test_path,
    transform=transform
)
test_loader = torch.utils.data.DataLoader(
    test_dataset,
    batch_size=32,
    num_workers=1,
    shuffle=True
)

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