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)