From dfc20600dae9c589b044d1c8462a77f19b13abf1 Mon Sep 17 00:00:00 2001 From: Chaithya G R Date: Wed, 10 Jan 2024 12:05:58 +0100 Subject: [PATCH] Update metrics.py --- modopt/math/metrics.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modopt/math/metrics.py b/modopt/math/metrics.py index 1b870e23..cf41a9c2 100644 --- a/modopt/math/metrics.py +++ b/modopt/math/metrics.py @@ -126,7 +126,7 @@ def ssim(test, ref, mask=None): test, ref, mask = _preprocess_input(test, ref, mask) test = move_to_cpu(test) - assim, ssim_value = compare_ssim(test, ref, full=True) + assim, ssim_value = compare_ssim(test, ref, full=True, data_range=1.0) if mask is None: return assim