bartduis commited on
Commit
51d5acd
·
verified ·
1 Parent(s): e277a17

Update eval_wrapper/sample_poses.py

Browse files
Files changed (1) hide show
  1. eval_wrapper/sample_poses.py +1 -1
eval_wrapper/sample_poses.py CHANGED
@@ -77,7 +77,7 @@ def sample_camera_poses(target: np.ndarray, inner_radius: float, outer_radius: f
77
  return np.stack(cameras)
78
 
79
 
80
- def pointmap_to_poses(pointmaps: torch.Tensor, n_poses: int, inner_radius: float = 1.1, outer_radius: float = 2.5, device: str = 'cuda',
81
  bb_mode: str='bb',run_octmae: bool = False) -> np.ndarray:
82
  """
83
  Samples `n_poses` camera poses uniformly on a sphere of given `radius` around `target`.
 
77
  return np.stack(cameras)
78
 
79
 
80
+ def pointmap_to_poses(pointmaps: torch.Tensor, n_poses: int, inner_radius: float = 1.1, outer_radius: float = 2.5, device: str = 'cpu',
81
  bb_mode: str='bb',run_octmae: bool = False) -> np.ndarray:
82
  """
83
  Samples `n_poses` camera poses uniformly on a sphere of given `radius` around `target`.