Drexubery commited on
Commit
84e37ca
1 Parent(s): b2bdc68
Files changed (1) hide show
  1. utils/pvd_utils.py +1 -1
utils/pvd_utils.py CHANGED
@@ -178,7 +178,7 @@ def generate_traj_txt(c2ws_anchor,H,W,fs,c,phi, theta, r,frame,device,viz_traj=F
178
  The camera coordinate sysmte in COLMAP is right-down-forward
179
  Pytorch3D is left-up-forward
180
  """
181
-
182
  if len(phi)>3:
183
  phis = txt_interpolation(phi,frame,mode='smooth')
184
  phis[0] = phi[0]
 
178
  The camera coordinate sysmte in COLMAP is right-down-forward
179
  Pytorch3D is left-up-forward
180
  """
181
+ c2ws_anchor = c2ws_anchor.to(device)
182
  if len(phi)>3:
183
  phis = txt_interpolation(phi,frame,mode='smooth')
184
  phis[0] = phi[0]