def run_s3dis()

in gss/selective_search_3d_run.py [0:0]

• 25 lines of code
• 11 McCabe index (conditional complexity)

def run_s3dis(names, masks):
    all_scan_names = [line.rstrip('/Annotations\n') for line in 
                open(os.path.join(FLAGS.data_path, 'meta/anno_paths.txt'))]
    if FLAGS.split == 'trainval':            
        scan_names = all_scan_names
    elif FLAGS.split in ['train', 'val']:
        if FLAGS.split == 'train':
            scan_names = []
            for scan in all_scan_names:
                if "Area_5" not in scan:
                    scan_names.append(scan)
        else:
            scan_names = []
            for scan in all_scan_names:
                if "Area_5" in scan:
                    scan_names.append(scan)
    else:
        raise ValueError('illegal split name')

    args = {'names': names, 'masks': masks}
    if FLAGS.n_proc != 0:
        with Pool(FLAGS.n_proc) as p:
            p.map(partial(_run_gss, args), scan_names)
    else:
        for scene_id in scan_names:
            _run_gss(args, scene_id)