updaaaaaaaaaaate

preprocess/preprocessor.py

@@ -67,7 +67,7 @@ def save_scene_data(root, scene, scene_idx=0, scene_total=1,file_type="txt"):
     target_mask_label = (0, 255, 0, 255)
     display_table_mask_label=(0, 0, 255, 255)
     random_downsample_N = 32768
-    voxel_size=0.002
+    voxel_size=0.003
     filter_degree = 75
     min_z = 0.2
     max_z = 0.5
@@ -135,7 +135,7 @@ 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/data/data/box_output"
+    root = r"/media/hofee/repository/full_data_output"
     # list_path = r"/media/hofee/repository/full_list.txt"
     # scene_list = []
     
@@ -144,7 +144,7 @@ if __name__ == "__main__":
     #         scene_list.append(line.strip())
     scene_list = os.listdir(root)
     from_idx = 0 # 1000
-    to_idx = len(scene_list) # 1500
+    to_idx = 700 # 1500
     print(scene_list)
 
 
@@ -153,7 +153,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, file_type="txt")
+        save_scene_data(root, scene, cnt, total, file_type="npy")
         cnt+=1
         end = time.time()
         print(f"Time cost: {end-start}")