From 5a8d9f66f252cc0d180cc8532c16b5ece386d147 Mon Sep 17 00:00:00 2001 From: jh Date: Wed, 17 Mar 2021 12:19:10 -0700 Subject: [PATCH] fixes loss range --- core/leras/layers/MsSsim.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/core/leras/layers/MsSsim.py b/core/leras/layers/MsSsim.py index 7843860..d45a599 100644 --- a/core/leras/layers/MsSsim.py +++ b/core/leras/layers/MsSsim.py @@ -24,7 +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) / 2.0 + return 1.0 - ms_ssim_val nn.MsSsim = MsSsim