def transform_test_img(self, sample): composed_transforms = transforms.Compose([ tr.Resize(224), tr.ToTensor(), tr.Normalize(mean=(0.324, 0.347, 0.307), std=(0.197, 0.185, 0.185)), ]) return composed_transforms(sample)
def transform_img(self, sample): composed_transforms = transforms.Compose([ tr.RandomFlip(), tr.RandomGaussianBlur(), tr.Resize(224), # tr.FixScaleCrop(224), tr.ToTensor(), tr.Normalize(mean=(0.324, 0.347, 0.307), std=(0.197, 0.185, 0.185)), ]) return composed_transforms(sample)