Decorrelated-Adversarial-Learning icon indicating copy to clipboard operation
Decorrelated-Adversarial-Learning copied to clipboard

resize added on train and valid data loader

Open hasithz opened this issue 2 years ago • 0 comments

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

hasithz avatar Nov 16 '23 08:11 hasithz