Update generate_metrics.py for usage of the unified PSNR signature

mlcommons · Jul 29, 2023 · af8c12d · af8c12d
1 parent 3ffbaa7
commit af8c12d
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/GANDLF/cli/generate_metrics.py b/GANDLF/cli/generate_metrics.py
@@ -14,7 +14,6 @@
     structural_similarity_index,
     mean_squared_error,
     peak_signal_noise_ratio,
-    peak_signal_noise_ratio_eps,
     mean_squared_log_error,
     mean_absolute_error,
     ncc_mean,
@@ -263,8 +262,8 @@ def __fix_2d_tensor(input_tensor):
                 gt_image_infill, output_infill
             ).item()
 
-            overall_stats_dict[current_subject_id]["psnr_range_eps"] = peak_signal_noise_ratio_eps(
-                gt_image_infill, output_infill
+            overall_stats_dict[current_subject_id]["psnr_eps"] = peak_signal_noise_ratio(
+                gt_image_infill, output_infill, epsilon=sys.float_info.epsilon
             ).item()
 
     pprint(overall_stats_dict)