Fix normal orientation loss

This commit is contained in:
Abhik Ahuja committed 2022-11-28 23:36:07 -08:00
1 parent d2e08ed1bc
commit 127d1d08e4
1 file changed
+1 -1
+1 -1
View File
@@ -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)