Decorrelated-Adversarial-Learning
Decorrelated-Adversarial-Learning copied to clipboard
resize added on train and valid data loader
Resize transformer block was added
ImageFolderWithAgeGroup(pat=self.dataset['pat'], pos=self.dataset['pos'], \
cutoffs=age_cutoffs, root=self.dataset['train_root'], transform=transforms.Compose(\
[transforms.RandomHorizontalFlip(p=0.5), transforms.ToTensor(), \
transforms.Normalize((0.5,0.5,0.5), (0.5,0.5,0.5)), transforms.Resize((96, 112))]))
this was added to the valid as well