search

neverUseThisName / Decorrelated-Adversarial-Learning

This is an unofficial implementation for the 2019 ICCV paper 'Decorrelated Adversarial Learning for Age-Invariant Face Recognition' from Tencent AI Lab
38 stars 15 forks source link

resize added on train and valid data loader #11

Open hasithz opened 9 months ago

hasithz commented 9 months ago

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