sync

preprocess/preprocessor.py

@@ -67,7 +67,7 @@ def save_scene_data(root, scene, scene_idx=0, scene_total=1,file_type="txt"):
     voxel_size=0.002
     filter_degree = 75
     min_z = 0.2
-    max_z = 0.45
+    max_z = 0.5
     
     ''' scan points '''
     display_table_info = DataLoadUtil.get_display_table_info(root, scene)
@@ -138,16 +138,16 @@ def save_scene_data(root, scene, scene_idx=0, scene_total=1,file_type="txt"):
 
 if __name__ == "__main__":
     #root = "/media/hofee/repository/new_data_with_normal"
-    root = r"/media/hofee/repository/new_data_with_normal"
-    list_path = r"/media/hofee/repository/full_list.txt"
-    scene_list = []
+    root = r"/media/hofee/data/tempdir/test_real_output"
+    # list_path = r"/media/hofee/repository/full_list.txt"
+    # scene_list = []
     
-    with open(list_path, "r") as f:
-        for line in f:
-            scene_list.append(line.strip())
-
-    from_idx = 1000
-    to_idx = 1500
+    # with open(list_path, "r") as f:
+    #     for line in f:
+    #         scene_list.append(line.strip())
+    scene_list = os.listdir(root)
+    from_idx = 0 # 1000
+    to_idx = 1 # 1500
 
 
     cnt = 0
@@ -155,7 +155,7 @@ if __name__ == "__main__":
     total = to_idx - from_idx
     for scene in scene_list[from_idx:to_idx]:
         start = time.time()
-        save_scene_data(root, scene, cnt, total, "npy")
+        save_scene_data(root, scene, cnt, total, "txt")
         cnt+=1
         end = time.time()
         print(f"Time cost: {end-start}")