def transform_val(self, sample): """transform for validation""" composed_transforms = transforms.Compose( [tr.Normalize(mean=self.mean, std=self.std), tr.ToTensor()]) return composed_transforms(sample)
def transform_train(self, sample): composed_transforms = transforms.Compose([ tr.RandomRescale(self.min_scale, self.max_scale, self.step_size), tr.RandomPadOrCrop(crop_height=self.crop_size, crop_width=self.crop_size, ignore_label=self.ignore_label, mean=self.mean), tr.RandomHorizontalFlip(), tr.Normalize(mean=self.mean, std=self.std), tr.ToTensor()]) return composed_transforms(sample)