From 4d80f0ceeefa13e91e14e9cdebec39232b5cb72e Mon Sep 17 00:00:00 2001
From: hturki <hturki@cs.cmu.edu>
Date: Tue, 25 Jul 2023 21:05:40 -0400
Subject: [PATCH] Use correct pixel area in volumetric sampler (#2268)

---
 nerfstudio/model_components/ray_samplers.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/nerfstudio/model_components/ray_samplers.py b/nerfstudio/model_components/ray_samplers.py
index c0b4b883d..e18c680ae 100644
--- a/nerfstudio/model_components/ray_samplers.py
+++ b/nerfstudio/model_components/ray_samplers.py
@@ -505,14 +505,13 @@ class VolumetricSampler(Sampler):
         if camera_indices is not None:
             camera_indices = camera_indices[ray_indices]
 
-        zeros = torch.zeros_like(origins[:, :1])
         ray_samples = RaySamples(
             frustums=Frustums(
                 origins=origins,
                 directions=dirs,
                 starts=starts[..., None],
                 ends=ends[..., None],
-                pixel_area=zeros,
+                pixel_area=ray_bundle[ray_indices].pixel_area,
             ),
             camera_indices=camera_indices,
         )