From 8c41a263b8c9aaba2ed242d0abaf03480a50eb1d Mon Sep 17 00:00:00 2001 From: Matthew Tancik Date: Fri, 27 Jan 2023 13:56:19 -0800 Subject: [PATCH] Fix normal orientation loss (#1298) Co-authored-by: Abhik Ahuja --- nerfstudio/model_components/losses.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nerfstudio/model_components/losses.py b/nerfstudio/model_components/losses.py index 9624ad02d..7fcfe0419 100644 --- a/nerfstudio/model_components/losses.py +++ b/nerfstudio/model_components/losses.py @@ -193,7 +193,7 @@ def orientation_loss( """ w = weights n = normals - v = viewdirs + v = viewdirs * -1 n_dot_v = (n * v[..., None, :]).sum(axis=-1) return (w[..., 0] * torch.fmin(torch.zeros_like(n_dot_v), n_dot_v) ** 2).sum(dim=-1)