修复小bug。

lartpang · Apr 9, 2024 · 8d8c86e · 8d8c86e
1 parent fa4e0a7
commit 8d8c86e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils/recorders/metric_recorder.py b/utils/recorders/metric_recorder.py
@@ -57,7 +57,7 @@ def round_w_zero_padding(x, bit_width):
     "dice": {"handler": py_sod_metrics.DICEHandler, "kwargs": gray_metric_kwargs},
     "specificity": {"handler": py_sod_metrics.SpecificityHandler, "kwargs": gray_metric_kwargs},
      #
-    "bif1": {"handler": py_sod_metrics.FmeasureHandler, "kwargs": dict(binary_metric_kwargs, beta=1)},
+    "bif1": {"handler": py_sod_metrics.FmeasureHandler, "kwargs": dict(**binary_metric_kwargs, beta=1)},
     "biprecision": {"handler": py_sod_metrics.PrecisionHandler, "kwargs": binary_metric_kwargs},
     "birecall": {"handler": py_sod_metrics.RecallHandler, "kwargs": binary_metric_kwargs},
     "biiou": {"handler": py_sod_metrics.IOUHandler, "kwargs": binary_metric_kwargs},
@@ -66,7 +66,7 @@ def round_w_zero_padding(x, bit_width):
 }
 GRAYSCALE_METRICS = ["em"] + [k for k in BINARY_METRIC_MAPPING.keys() if not k.startswith("bi")]
 SUPPORTED_METRICS = ["mae", "em", "sm", "wfm", "msiou"] + sorted(BINARY_METRIC_MAPPING.keys())
-# fmt: off
+# fmt: on
 
 
 class GrayscaleMetricRecorder: