From edb54602da44e493cd3483d62a22a0a690cb6690 Mon Sep 17 00:00:00 2001 From: Koke_Cacao Date: Tue, 28 Feb 2023 02:40:17 +0800 Subject: [PATCH] Bug fix for SHRenderer() (#1526) The current SHRenderer does not apply the activation function correctly due to a typo. Now it's fixed. --- nerfstudio/model_components/renderers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nerfstudio/model_components/renderers.py b/nerfstudio/model_components/renderers.py index 3d01bddea..5e488bc35 100644 --- a/nerfstudio/model_components/renderers.py +++ b/nerfstudio/model_components/renderers.py @@ -183,7 +183,7 @@ class SHRenderer(nn.Module): rgb = torch.sum(sh, dim=-1) + 0.5 # [..., num_samples, 3] if self.activation is not None: - self.activation(rgb) + rgb = self.activation(rgb) if not self.training: rgb = torch.nan_to_num(rgb)