def transform_val(self, sample): composed_transforms = transforms.Compose([ # tr.FixScaleCrop(crop_size=self.args.crop_size), tr.FixedNoMaskResize(size=self.args.crop_size), tr.Normalize(mean=(0.485, 0.456, 0.406), std=(0.229, 0.224, 0.225)), tr.ToTensor()]) return composed_transforms(sample)
def transform_tr(self, sample): composed_transforms = transforms.Compose([ tr.FixedNoMaskResize(size=self.args.crop_size), tr.RandomColorJeter(0.3, 0.3, 0.3, 0.3), tr.RandomHorizontalFlip(), # tr.RandomScaleCrop(base_size=self.args.base_size, crop_size=self.args.crop_size), tr.RandomGaussianBlur(), tr.Normalize(mean=(0.485, 0.456, 0.406), std=(0.229, 0.224, 0.225)), tr.ToTensor()]) return composed_transforms(sample)