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)