Update metrics.py

CEA-COSMIC · Jan 10, 2024 · dfc2060 · dfc2060
1 parent 3bae0ac
commit dfc2060
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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