예제에서 사용할 이미지 데이터에 대한 전처리 방법을 정의합니다.
코드 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))