From 5e08b622da26e79159fea3758791130c640bf7d0 Mon Sep 17 00:00:00 2001 From: Junchen Liu Date: Wed, 24 Apr 2024 09:59:31 +0800 Subject: [PATCH] fix viewdir checking in ngp.py --- examples/radiance_fields/ngp.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/radiance_fields/ngp.py b/examples/radiance_fields/ngp.py index 44b81c5..3a889ea 100644 --- a/examples/radiance_fields/ngp.py +++ b/examples/radiance_fields/ngp.py @@ -212,8 +212,8 @@ def forward( assert ( positions.shape == directions.shape ), f"{positions.shape} v.s. {directions.shape}" - density, embedding = self.query_density(positions, return_feat=True) - rgb = self._query_rgb(directions, embedding=embedding) + density, embedding = self.query_density(positions, return_feat=True) + rgb = self._query_rgb(directions, embedding=embedding) return rgb, density # type: ignore