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: