tink/nerfstudio
4
0 Fork 0
mirror of https://github.com/nerfstudio-project/nerfstudio.git synced 2026-11-13 11:24:00 +08:00
Code Activity

Fix depth nerfacto exporting (#1453)

Browse Source
This commit is contained in:
Matthew Tancik authored and GitHub committed 2023-02-19 13:53:01 -08:00
1 parent 73f0f5a16a
commit ceda935e64
1 file changed
+2 -1
Download Patch File Download Diff File Expand all files Collapse all files
nerfstudio/models/depth_nerfacto.py
+2 -1
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):