def transform_pair_val(self, sample):
composed_transforms = transforms.Compose([
tr.FixScaleCrop(400),
tr.HorizontalFlip(),
tr.GaussianBlur(),
tr.Normalize(mean=self.source_dist['mean'], std=self.source_dist['std'], if_pair=True),
tr.ToTensor(if_pair=True),
])
return composed_transforms(sample)
def transform_pair_train(self, sample):
composed_transforms = transforms.Compose([
tr.RandomHorizontalFlip(),
tr.RandomScaleCrop(base_size=400, crop_size=400, fill=0),
tr.HorizontalFlip(),
tr.GaussianBlur(),
tr.Normalize(mean=self.source_dist['mean'], std=self.source_dist['std'], if_pair=True),
tr.ToTensor(if_pair=True),
])
return composed_transforms(sample)
def transform_pair_val(self, sample):
composed_transforms = transforms.Compose([
tr.FixScaleCrop(400),
#tr.Normalize(mean=(0.485, 0.456, 0.406), std=(0.229, 0.224, 0.225)),
tr.HorizontalFlip(),
tr.GaussianBlur(),
tr.Normalize(if_pair=True),
tr.ToTensor(if_pair=True),
])
return composed_transforms(sample)