From ebbebdab4c602fc9b993a6ab7095f9adc771c1d5 Mon Sep 17 00:00:00 2001
From: mattstrong-stanford
 <158235488+mattstrong-stanford@users.noreply.github.com>
Date: Wed, 31 Jan 2024 11:48:08 -0800
Subject: [PATCH] Have predicted depth as expected_depth so gradients can be
 computed (#2852)

Update depth_nerfacto.py
---
 nerfstudio/models/depth_nerfacto.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/nerfstudio/models/depth_nerfacto.py b/nerfstudio/models/depth_nerfacto.py
index e439ce677..1a47ab163 100644
--- a/nerfstudio/models/depth_nerfacto.py
+++ b/nerfstudio/models/depth_nerfacto.py
@@ -95,7 +95,7 @@ class DepthNerfactoModel(NerfactoModel):
                         weights=outputs["weights_list"][i],
                         ray_samples=outputs["ray_samples_list"][i],
                         termination_depth=termination_depth,
-                        predicted_depth=outputs["depth"],
+                        predicted_depth=outputs["expected_depth"],
                         sigma=sigma,
                         directions_norm=outputs["directions_norm"],
                         is_euclidean=self.config.is_euclidean_depth,