From 5f10e3c67e4296066d78263c34a81ba95b894202 Mon Sep 17 00:00:00 2001 From: jh Date: Wed, 17 Mar 2021 11:32:05 -0700 Subject: [PATCH] revert changes --- core/leras/layers/MsSsim.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/core/leras/layers/MsSsim.py b/core/leras/layers/MsSsim.py index cb48294..7843860 100644 --- a/core/leras/layers/MsSsim.py +++ b/core/leras/layers/MsSsim.py @@ -24,6 +24,7 @@ class MsSsim(nn.LayerBase): ms_ssim_val = tf.image.ssim_multiscale(y_true_t, y_pred_t, max_val, power_factors=self.power_factors, filter_size=self.kernel_size) # ssim_multiscale returns values in range [0, 1] (where 1 is completely identical) # subtract from 1 to get loss - return 1.0 - ms_ssim_val + return (1.0 - ms_ssim_val) / 2.0 + nn.MsSsim = MsSsim