Use tuple for datat range & fixed the missing range in the psnr_01_ep…

…s case
mlcommons · Aug 10, 2023 · 5690df7 · 5690df7
1 parent 1d492aa
commit 5690df7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/GANDLF/cli/generate_metrics.py b/GANDLF/cli/generate_metrics.py
@@ -301,14 +301,14 @@ def __percentile_clip(input_tensor, reference_tensor=None, p_min=0.5, p_max=99.5
                 overall_stats_dict[current_subject_id][
                     "psnr_01"
                 ] = peak_signal_noise_ratio(
-                    gt_image_infill, output_infill, data_range=1.0
+                    gt_image_infill, output_infill, data_range=(0,1)
                 ).item()
 
                 # same as above but with epsilon for robustness
                 overall_stats_dict[current_subject_id][
                     "psnr_01_eps"
                 ] = peak_signal_noise_ratio(
-                    gt_image_infill, output_infill, epsilon=sys.float_info.epsilon
+                    gt_image_infill, output_infill, data_range=(0,1), epsilon=sys.float_info.epsilon
                 ).item()
 
     pprint(overall_stats_dict)