tink/nerfstudio
4
0
Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-10 18:17:31 +08:00
Code

Zipnerf point cloud and tsdf mesh export support (#3429)

Browse Source 
Co-authored-by: J.Y. <132313008+jb-ye@users.noreply.github.com>
This commit is contained in:
Aayush Gupta 2024-09-24 12:28:18 -07:00 committed by GitHub
parent 516fd7c9ac
commit e6941aca47
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
nerfstudio/scripts/exporter.py
View File

@ -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: