Skip to content

Commit

Permalink
add output type to _align_points_to_target_plane
Browse files Browse the repository at this point in the history
  • Loading branch information
hoanhle committed Aug 26, 2024
1 parent 867a4da commit 5dd54b1
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions nerfstudio/data/dataparsers/nerfstudio_dataparser.py
Original file line number Diff line number Diff line change
Expand Up @@ -494,8 +494,12 @@ def _load_3D_points(self, ply_file_path: Path, transform_matrix: torch.Tensor, s

@staticmethod
def _align_points_to_target_plane(
points: torch.Tensor, target_normal: torch.Tensor, threshold: float = 1.0, max_iterations: int = 5
):
points: torch.Tensor,
target_normal: Tuple[float, float, float],
target_point: Tuple[float, float, float],
threshold: float = 1.0,
max_iterations: int = 5,
) -> Tuple[torch.Tensor, torch.Tensor]:
"""Aligns a set of 3D points (in homogeneous coordinates) to a target plane defined by its normal vector.
Args:
Expand Down

0 comments on commit 5dd54b1

Please sign in to comment.