solve merge

runners/strategy_generator.py

@@ -86,13 +86,17 @@ class StrategyGenerator(Runner):
             nrm_path = os.path.join(root,scene_name, "nrm", f"{frame_idx}.npy")
             idx_path = os.path.join(root,scene_name, "scan_points_indices", f"{frame_idx}.npy")
             pts = np.load(pts_path)
-            if pts.shape[0] == 0:
-                nrm = np.zeros((0,3))
-            else:
-                nrm = np.load(nrm_path)
-            indices = np.load(idx_path) 
+            if self.compute_with_normal:
+                if pts.shape[0] == 0:
+                    nrm = np.zeros((0,3))
+                else:
+                    nrm = np.load(nrm_path)
+                nrm_list.append(nrm)
+            
             pts_list.append(pts)
-            nrm_list.append(nrm)
+            
+            indices = np.load(idx_path) 
+            
             scan_points_indices_list.append(indices)
             if pts.shape[0] > 0:
                 non_zero_cnt += 1