From 127d1d08e4328a11c581a8b5ef03e7e97ae0e379 Mon Sep 17 00:00:00 2001 From: Abhik Ahuja Date: Mon, 28 Nov 2022 23:36:07 -0800 Subject: [PATCH] Fix normal orientation loss --- 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 1946e4abc..a5529e5c8 100644 --- a/nerfstudio/model_components/losses.py +++ b/nerfstudio/model_components/losses.py @@ -182,7 +182,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)