diff --git a/anomalib/utils/metrics/adaptive_threshold.py b/anomalib/utils/metrics/adaptive_threshold.py index 37c85d68d1..eeeec4fbed 100644 --- a/anomalib/utils/metrics/adaptive_threshold.py +++ b/anomalib/utils/metrics/adaptive_threshold.py @@ -44,3 +44,7 @@ def compute(self) -> torch.Tensor: else: self.value = thresholds[torch.argmax(f1_score)] return self.value + + def reset(self) -> None: + """Reset the metric.""" + self.precision_recall_curve.reset() diff --git a/anomalib/utils/metrics/optimal_f1.py b/anomalib/utils/metrics/optimal_f1.py index 44f194a770..403e774c6f 100644 --- a/anomalib/utils/metrics/optimal_f1.py +++ b/anomalib/utils/metrics/optimal_f1.py @@ -40,3 +40,7 @@ def compute(self) -> torch.Tensor: self.threshold = thresholds[torch.argmax(f1_score)] optimal_f1_score = torch.max(f1_score) return optimal_f1_score + + def reset(self) -> None: + """Reset the metric.""" + self.precision_recall_curve.reset()