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

Fix depth nerfacto exporting (#1453)

Browse Source
This commit is contained in:
Matthew Tancik 2023-02-19 13:53:01 -08:00 committed by GitHub
parent 73f0f5a16a
commit ceda935e64
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
nerfstudio/models/depth_nerfacto.py
View File

@ -70,7 +70,8 @@ class DepthNerfactoModel(NerfactoModel):
def get_outputs(self, ray_bundle: RayBundle):
outputs = super().get_outputs(ray_bundle)
outputs["directions_norm"] = ray_bundle.metadata["directions_norm"]
if ray_bundle.metadata is not None and "directions_norm" in ray_bundle.metadata:
outputs["directions_norm"] = ray_bundle.metadata["directions_norm"]
return outputs
def get_metrics_dict(self, outputs, batch):