다음은 데이터셋 전처리를 위한 항목들을 정의합니다. 전처리에는 데이터 크기 조정 및 데이터 정규화(분포 조정) 등이 포함됩니다.
코드 8-21 데이터셋 전처리
train_transform = transforms.Compose([
transforms.Resize((224, 224)),
transforms.RandomHorizontalFlip(),
transforms.RandomVerticalFlip(),
transforms.ToTensor(),
transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
])
val_transform = transforms.Compose([
transforms.Resize((224, 224)),
transforms.ToTensor(),
transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])
])
예제에서 사용하는 데이터셋은 핫도그와 핫도그가 아닌 이미지들을 사용합니다. 데이터셋은 https://www.kaggle.com/dansbecker/hot-dog-not-hot-dog에서 내려받을 수 있지만 이것 역시 ‘Food 101 dataset’을 이용한 것입니다.