wassname/keras-contrib
1
0
Fork 0
mirror of https://github.com/wassname/keras-contrib.git synced 2026-06-27 16:10:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix PEP8 error

Browse Source
This commit is contained in:
Michael Oliver
2017-06-08 08:37:20 -07:00
committed by GitHub
parent 3560fb7b6e
commit e2da54b993
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
keras_contrib/losses/dssim.py
+1 -1
View File
@@ -53,7 +53,7 @@ class DSSIMObjective():
var_true = K.var(patches_true, axis=-1)
var_pred = K.var(patches_pred, axis=-1)
# Get std dev
covar_true_pred = K.mean(patches_true*patches_pred, axis=-1) - u_true*u_pred
covar_true_pred = K.mean(patches_true * patches_pred, axis=-1) - u_true * u_pred
ssim = (2 * u_true * u_pred + self.c1) * (2 * covar_true_pred + self.c2)
denom = (K.square(u_true) + K.square(u_pred) + self.c1) * (var_pred + var_true + self.c2)