def transforms(self):
train_transforms = Compose([
RandomScale(scale=self.scale),
RandomCrop(crop_size=self.size),
RandomFlip()
])
val_transforms = Compose([Resize(size=self.size)])
return train_transforms, val_transforms
def transforms(self):
train_transforms = Compose([
# RandomScale(scale=self.scale),
RandomCrop(crop_size=self.size),
RandomFlip(),
Normalize() if self.normalize else Tensorize()
])
val_transforms = Compose([
Resize(size=self.size),
Normalize() if self.normalize else Tensorize()
])
return train_transforms, val_transforms
