Method __init__

dataloader/data_augmentation.py:73–88

(self, img_size, mode=0, is_train=False)

class Augment_SPH_Data():
    def __init__(self, img_size, mode=0, is_train=False):
        if is_train:
            if mode == 0:
                transform = [transforms.ToTensor(), 
                             transforms.Normalize([0.5], [0.5])]
            elif mode == 1:
                transform = [SphRandomRotate(p=1, img_size=img_size),
                             transforms.ToTensor(),
                             transforms.Normalize([0.5], [0.5])]
            else:
                raise NotImplementedError(f'Uncognized data augmentation mode')
        else:
            transform = [transforms.ToTensor(),
                         transforms.Resize(img_size, Image.BICUBIC),
                         transforms.Normalize([0.5], [0.5])]
        self.transform = transforms.Compose(transform)
    
    def __call__(self, input):
        output = self.transform(input)