diff --git a/src/anomalib/metrics/__init__.py b/src/anomalib/metrics/__init__.py index 544e6fbf6f..151e841085 100644 --- a/src/anomalib/metrics/__init__.py +++ b/src/anomalib/metrics/__init__.py @@ -12,6 +12,7 @@ import torchmetrics from omegaconf import DictConfig, ListConfig +from . import per_image from .anomaly_score_distribution import AnomalyScoreDistribution from .aupr import AUPR from .aupro import AUPRO @@ -30,6 +31,7 @@ "ManualThreshold", "MinMax", "PRO", + "per_image", ] logger = logging.getLogger(__name__)