Ejemplo n.º 1
0
    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)
Ejemplo n.º 2
0
	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)