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

update gf_view_finder

Browse Source
This commit is contained in:
hofee
2024-09-02 23:47:52 +08:00
parent 2fcfcd1966
commit e0fb9a7617
7 changed files with 78 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
modules/func_lib/samplers.py
View File

@@ -88,8 +88,8 @@ def cond_ode_sampler(
x = mean_x
num_steps = xs.shape[0]
xs = xs.reshape(batch_size * num_steps, -1)
xs = PoseUtil.normalize_rotation(xs, pose_mode)
xs = xs.reshape(batch_size*num_steps, -1)
xs[:, :-3] = PoseUtil.normalize_rotation(xs[:, :-3], pose_mode)
xs = xs.reshape(num_steps, batch_size, -1)
x = PoseUtil.normalize_rotation(x, pose_mode)
x[:, :-3] = PoseUtil.normalize_rotation(x[:, :-3], pose_mode)
return xs.permute(1, 0, 2), x