From 6b78f1cbc3a0ad4d65bb3c7a428f4ccdab169c18 Mon Sep 17 00:00:00 2001 From: jh Date: Fri, 12 Mar 2021 08:18:56 -0800 Subject: [PATCH] fix: misnamed arg for tf.image.ssim_multiscale --- core/leras/ops/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/core/leras/ops/__init__.py b/core/leras/ops/__init__.py index b7f1bab..398679c 100644 --- a/core/leras/ops/__init__.py +++ b/core/leras/ops/__init__.py @@ -325,7 +325,7 @@ def ms_ssim(img1, img2, resolution, kernel_size=11, k1=0.01, k2=0.03, max_value= if sum(power_factors) < 1.0: power_factors = [x/sum(power_factors) for x in power_factors] - ms_ssim_val = tf.image.ssim_multiscale(img1, img2, max_value=max_value, power_factors=power_factors, + ms_ssim_val = tf.image.ssim_multiscale(img1, img2, max_val=max_value, power_factors=power_factors, filter_size=kernel_size, k1=k1, k2=k2) loss = (1.0 - ms_ssim_val) / 2.0