Method __init__

dataloader/data_augmentation.py:22–39

(self, mode=0, is_train=False)

class Augment_Point_Data():
    def __init__(self, mode=0, is_train=False):
        if is_train:
            if mode == 0:
                transform = [transforms.ToTensor()]
            elif mode == 1:
                transform = [transforms.ToTensor(), 
                             JitterPoints(sigma=0.001, clip=0.002), 
                             RemoveRandomPoints(r=(0.0, 0.1)),
                             RandomTranslation(max_delta=0.01), 
                             RemoveRandomBlock(p=0.4),
                             RandomRotation(axis=np.array([0,0,1]),
                                            max_theta=15,
                                            max_theta2=None)]
            else:
                raise NotImplementedError(f'Uncognized data augmentation mode')
        else:
            transform = [transforms.ToTensor()]
        self.transform = transforms.Compose(transform)
    
    def __call__(self, input):
        output = self.transform(input)