Skip to content

Commit

Permalink
Zipnerf point cloud and tsdf mesh export support (#3429)
Browse files Browse the repository at this point in the history
Co-authored-by: J.Y. <[email protected]>
  • Loading branch information
aayushg55 and jb-ye authored Sep 24, 2024
1 parent 516fd7c commit e6941ac
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion nerfstudio/scripts/exporter.py
Original file line number Diff line number Diff line change
Expand Up @@ -73,8 +73,13 @@ def validate_pipeline(normal_method: str, normal_output_name: str, pipeline: Pip
directions = torch.ones_like(origins)
pixel_area = torch.ones_like(origins[..., :1])
camera_indices = torch.zeros_like(origins[..., :1])
metadata = {"directions_norm": torch.linalg.vector_norm(directions, dim=-1, keepdim=True)}
ray_bundle = RayBundle(
origins=origins, directions=directions, pixel_area=pixel_area, camera_indices=camera_indices
origins=origins,
directions=directions,
pixel_area=pixel_area,
camera_indices=camera_indices,
metadata=metadata,
)
outputs = pipeline.model(ray_bundle)
if normal_output_name not in outputs:
Expand Down

0 comments on commit e6941ac

Please sign in to comment.