OSSForks/faceswap
1
0
Fork 0
mirror of https://github.com/zebrajr/faceswap.git synced 2025-12-06 00:20:09 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

bugfix - losses - DSSIM - Don't reduce mean on output

Browse Source
This commit is contained in:
torzdf 2020-08-21 15:29:27 +01:00
parent 619bd415aa
commit 877b90b60a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lib/model/losses_tf.py
View File

@ -136,7 +136,7 @@ class DSSIMObjective(tf.keras.losses.Loss):
denom = (K.square(u_true) + K.square(u_pred) + self.c_1) * (
var_pred + var_true + self.c_2)
ssim /= denom # no need for clipping, c_1 + c_2 make the denorm non-zero
return K.mean((1.0 - ssim) / 2.0)
return (1.0 - ssim) / 2.0
@staticmethod
def _preprocess_padding(padding):