Update eval_wrapper/sample_poses.py
Browse files
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 = '
|
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`.
|