From 703c790e67d110e047a1750fd2d6f34cb5085fc1 Mon Sep 17 00:00:00 2001 From: jh Date: Sat, 13 Mar 2021 14:39:36 -0800 Subject: [PATCH] kernel size fix --- 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 b8d4c45..9aebf8f 100644 --- a/core/leras/layers/MsSsim.py +++ b/core/leras/layers/MsSsim.py @@ -20,7 +20,7 @@ class MsSsim(nn.LayerBase): y_true_t = tf.transpose(tf.cast(y_true, tf.float32), [0, 2, 3, 1]) y_pred_t = tf.transpose(tf.cast(y_pred, tf.float32), [0, 2, 3, 1]) - loss = tf.image.ssim_multiscale(y_true_t, y_pred_t, max_val, power_factors=self.power_factors) + loss = tf.image.ssim_multiscale(y_true_t, y_pred_t, max_val, power_factors=self.power_factors, filter_size=self.kernel_size) return (1.0 - loss) / 2.0 nn.MsSsim = MsSsim