def __init__()

in wypr/dataset/s3dis/s3dis.py [0:0]

• 35 lines of code
• 10 McCabe index (conditional complexity)

    def __init__(self, split_set='train', num_points=20000,
                use_color=False, use_height=False, augment=False, 
                use_normal=False, precomputed_prop="", sampling_method='random'):

        self.CONFIG = S3DISDatasetConfig()
        self.data_path = os.path.join(BASE_DIR, 's3dis_all_points')
        all_scan_names = [line.rstrip('/Annotations\n') for line in open('meta/anno_paths.txt')]
        
        if split_set=='all':            
            self.scan_names = all_scan_names
        elif split_set in ['train', 'val', 'test']:
            if split_set == 'train':
                self.scan_names = []
                for scan in all_scan_names:
                    if "Area_5" not in scan:
                        self.scan_names.append(scan)
            else:
                self.scan_names = []
                for scan in all_scan_names:
                    if "Area_5" in scan:
                        self.scan_names.append(scan)
        elif split_set=='debug':   
            self.scan_names = all_scan_names[:10]
        else:
            print('illegal split name')
            return

        num_scans = len(all_scan_names)
        print('kept {} scans out of {}'.format(len(self.scan_names), num_scans))
        
        self.split_set = split_set
        self.num_points_sampled = num_points
        self.use_color = use_color        
        self.use_height = use_height
        self.use_normal = use_normal
        self.augment = augment
        self.sampling_method = sampling_method

        # proposals
        self.precomputed_prop = precomputed_prop 
        self.prop_path = os.path.join(BASE_DIR, 'proposals')