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

    코드 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))
    신간 소식 구독하기
    뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.