diff --git a/data_utils/collect_indoor3d_data.py b/data_utils/collect_indoor3d_data.py index 81ecdbee6..09f371303 100644 --- a/data_utils/collect_indoor3d_data.py +++ b/data_utils/collect_indoor3d_data.py @@ -6,10 +6,10 @@ ROOT_DIR = os.path.dirname(BASE_DIR) sys.path.append(BASE_DIR) -anno_paths = [line.rstrip() for line in open(os.path.join(BASE_DIR, 'meta/anno_paths.txt'))] -anno_paths = [os.path.join(DATA_PATH, p) for p in anno_paths] +anno_paths = [line.rstrip() for line in open(os.path.join(BASE_DIR, 'meta', 'anno_paths.txt'))] +anno_paths = [os.path.join(DATA_PATH, p.replace('/', os.sep)) for p in anno_paths] -output_folder = os.path.join(ROOT_DIR, 'data/stanford_indoor3d') +output_folder = os.path.join(ROOT_DIR, 'data', 's3dis', 'stanford_indoor3d') if not os.path.exists(output_folder): os.mkdir(output_folder) @@ -17,7 +17,7 @@ for anno_path in anno_paths: print(anno_path) try: - elements = anno_path.split('/') + elements = anno_path.split(os.sep) out_filename = elements[-3]+'_'+elements[-2]+'.npy' # Area_1_hallway_1.npy collect_point_label(anno_path, os.path.join(output_folder, out_filename), 'numpy') except: