hofee/nbv_reconstruction
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
hofee
2024-10-28 16:48:34 +00:00
parent 5c56dae24f
commit 49bcf203a8
3 changed files with 15 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
runners/strategy_generator.py
View File

@@ -85,14 +85,18 @@ class StrategyGenerator(Runner):
pts_path = os.path.join(root,scene_name, "pts", f"{frame_idx}.npy")
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)
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)
indices = np.load(idx_path)
pts_list.append(pts)
nrm_list.append(nrm)
scan_points_indices_list.append(indices)
if pts.shape[0] > 0:
non_zero_cnt += 1