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