From 473560e399b266d83e9f6415423673b34be90334 Mon Sep 17 00:00:00 2001
From: JonathonLuiten <jonoluiten@gmail.com>
Date: Wed, 12 Jul 2023 20:35:43 -0400
Subject: [PATCH] Update __init__.py

---
 gaussian_renderer/__init__.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/gaussian_renderer/__init__.py b/gaussian_renderer/__init__.py
index 7ca4cf3..dc23335 100644
--- a/gaussian_renderer/__init__.py
+++ b/gaussian_renderer/__init__.py
@@ -81,7 +81,7 @@ def render(viewpoint_camera, pc : GaussianModel, pipe, bg_color : torch.Tensor,
         colors_precomp = override_color
 
     # Rasterize visible Gaussians to image, obtain their radii (on screen). 
-    rendered_image, radii = rasterizer(
+    rendered_image, radii, depth = rasterizer(
         means3D = means3D,
         means2D = means2D,
         shs = shs,
@@ -95,5 +95,6 @@ def render(viewpoint_camera, pc : GaussianModel, pipe, bg_color : torch.Tensor,
     # They will be excluded from value updates used in the splitting criteria.
     return {"render": rendered_image,
             "viewspace_points": screenspace_points,
-            "visibility_filter" : radii > 0,
-            "radii": radii}
+            "visibility_filter": radii > 0,
+            "radii": radii,
+           "depth": depth}