Skip to content

resize added on train and valid data loader#11

Open
hasithz wants to merge 2 commits intoneverUseThisName:masterfrom
hasithz:resize-add
Open

resize added on train and valid data loader#11
hasithz wants to merge 2 commits intoneverUseThisName:masterfrom
hasithz:resize-add

Conversation

@hasithz
Copy link

@hasithz hasithz commented Nov 16, 2023

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant