From e6941aca4746a7970ac08f6761af5d49d6785b65 Mon Sep 17 00:00:00 2001 From: Aayush Gupta <19579293+aayushg55@users.noreply.github.com> Date: Tue, 24 Sep 2024 12:28:18 -0700 Subject: [PATCH] Zipnerf point cloud and tsdf mesh export support (#3429) Co-authored-by: J.Y. <132313008+jb-ye@users.noreply.github.com> --- nerfstudio/scripts/exporter.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/nerfstudio/scripts/exporter.py b/nerfstudio/scripts/exporter.py index ca019850f..650eaebff 100644 --- a/nerfstudio/scripts/exporter.py +++ b/nerfstudio/scripts/exporter.py @@ -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: