luni/cvpr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

use mean instead of sum

Browse Source
This commit is contained in:
luni
2022-06-16 15:28:23 +00:00
parent b0b42c1bd4
commit e1602567d6
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
losses.py
+2 -2
View File
@@ -124,7 +124,7 @@ def collision_loss(V, B, mask, thr=.004):
D = V - tf.gather_nd(B, idx) D = V - tf.gather_nd(B, idx)
b_vn = tf.gather_nd(B_vn, idx) b_vn = tf.gather_nd(B_vn, idx)
dot = tf.einsum('abc,abc->ab', D, b_vn) dot = tf.einsum('abc,abc->ab', D, b_vn)
loss = tf.reduce_sum((tf.minimum(dot - thr, 0) * mask) ** 2) loss = tf.reduce_mean((tf.minimum(dot - thr, 0) * mask) ** 2)
return loss return loss
@@ -147,7 +147,7 @@ def edge_loss(x, y, mask):
y_e = tf.sqrt(y_e + 1e-9) y_e = tf.sqrt(y_e + 1e-9)
d_e = y_e - x_e d_e = y_e - x_e
d_e = d_e * e_mask d_e = d_e * e_mask
loss = tf.reduce_sum(d_e ** 2) loss = tf.reduce_mean(d_e ** 2)
return loss return loss