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

solve merge

Browse Source
This commit is contained in:
hofee
2025-04-11 20:10:56 +00:00
parent be835aded4 7c7f071f95
commit ad7a1c9cdf
23 changed files with 1757 additions and 119 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
core/pipeline.py
View File

@@ -75,6 +75,8 @@ class NBVReconstructionPipeline(nn.Module):
def forward_test(self, data):
main_feat = self.get_main_feat(data)
repeat_num = data.get("repeat_num", 1)
main_feat = main_feat.repeat(repeat_num, 1)
estimated_delta_rot_9d, in_process_sample = self.view_finder.next_best_view(
main_feat
)
@@ -90,6 +92,8 @@ class NBVReconstructionPipeline(nn.Module):
] # List(B): Tensor(S x 9)
scanned_pts_mask_batch = data["scanned_pts_mask"] # List(B): Tensor(S x N)
scanned_pts_mask_batch = data["scanned_pts_mask"] # List(B): Tensor(N)
device = next(self.parameters()).device
embedding_list_batch = []