먼저 모델 학습에 필요한 데이터셋의 전처리(예 텐서 변환)가 필요합니다.
코드 6-2 이미지 데이터셋 전처리
class ImageTransform():
def __init__(self, resize, mean, std):
self.data_transform = {
'train': transforms.Compose([
transforms.RandomResizedCrop(resize, scale=(0.5,1.0)),
transforms.RandomHorizontalFlip(),
transforms.ToTensor(),
transforms.Normalize(mean, std)
]), ------ ①
'val': transforms.Compose([
transforms.Resize(256),
transforms.CenterCrop(resize),
transforms.ToTensor(),
transforms.Normalize(mean, std)
])
}
def __call__(self, img, phase): ------ ②
return self.data_transform[phase](img)