Merge branch 'master' of https://git.hofee.top/hofee/nbv_reconstruction

preprocess/preprocessor.py

@@ -78,7 +78,11 @@ def save_scene_data(root, scene, scene_idx=0, scene_total=1,file_type="txt"):
     ''' read frame data(depth|mask|normal) '''
     frame_num = DataLoadUtil.get_scene_seq_length(root, scene)
     for frame_id in range(frame_num):
+    
+        
         print(f"[scene({scene_idx}/{scene_total})|frame({frame_id}/{frame_num})]Processing {scene} frame {frame_id}")
+        if frame_id != 126:
+            continue
         path = DataLoadUtil.get_path(root, scene, frame_id)
         cam_info = DataLoadUtil.load_cam_info(path, binocular=True)
         depth_L, depth_R = DataLoadUtil.load_depth(
@@ -148,6 +152,8 @@ if __name__ == "__main__":
 
     from_idx = 0
     to_idx = len(scene_list)
+
+
     cnt = 0
     import time
     total = to_idx - from_idx
@@ -156,4 +162,4 @@ if __name__ == "__main__":
         save_scene_data(root, scene, cnt, total, file_type="npy")
         cnt+=1
         end = time.time()
-        print(f"Time cost: {end-start}")
+        print(f"Time cost: {end-start}")