collect_indoor3d_data.py

WangYueFt · Feb 4, 2018 · 9dce447 · 9dce447
1 parent 18fcede
commit 9dce447
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/sem_seg/collect_indoor3d_data.py b/sem_seg/collect_indoor3d_data.py
@@ -10,14 +10,14 @@
 
 output_folder = os.path.join(ROOT_DIR, 'data/stanford_indoor3d') 
 if not os.path.exists(output_folder):
-    os.mkdir(output_folder)
+  os.mkdir(output_folder)
 
 # Note: there is an extra character in the v1.2 data in Area_5/hallway_6. It's fixed manually.
 for anno_path in anno_paths:
-    print(anno_path)
-    try:
-        elements = anno_path.split('/')
-        out_filename = elements[-3]+'_'+elements[-2]+'.npy'
-        indoor3d_util.collect_point_label(anno_path, os.path.join(output_folder, out_filename), 'numpy')
-    except:
-        print(anno_path, 'ERROR!!')
+  print(anno_path)
+  try:
+    elements = anno_path.split('/')
+    out_filename = elements[-3]+'_'+elements[-2]+'.npy'
+    indoor3d_util.collect_point_label(anno_path, os.path.join(output_folder, out_filename), 'numpy')
+  except:
+    print(anno_path, 'ERROR!!')