upd

preprocess/pack_upload_data.py

@@ -29,8 +29,8 @@ def pack_all_scenes(root, scene_list, output_dir):
         pack_scene_data(root, scene, output_dir)
 
 if __name__ == "__main__":
-    root = r"H:\AI\Datasets\nbv_rec_part2"
-    output_dir = r"H:\AI\Datasets\upload_part2"
+    root = r"/media/hofee/repository/data_part_1"
+    output_dir = r"/media/hofee/repository/upload_part1"
     scene_list = os.listdir(root)
     from_idx = 0
     to_idx = len(scene_list)

preprocess/preprocessor.py

@@ -164,10 +164,10 @@ 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"H:\AI\Datasets\nbv_rec_part2"
+    root = "/media/hofee/data/data/new_testset"
     scene_list = os.listdir(root)
     from_idx = 0 # 1000
-    to_idx = 600 # 1500
+    to_idx = len(scene_list) # 1500
 
 
     cnt = 0
@@ -179,7 +179,11 @@ if __name__ == "__main__":
             print(f"Scene {scene} has been processed")
             cnt+=1
             continue
-        save_scene_data(root, scene, cnt, total, file_type="npy")
+        try:
+            save_scene_data(root, scene, cnt, total, file_type="npy")
+        except Exception as e:
+            print(f"Error occurred when processing scene {scene}")
+            print(e)
         cnt+=1
         end = time.time()
         print(f"Time cost: {end-start}")