use clamp instead of min/max

openvinotoolkit · Jan 14, 2025 · 1feebc0 · 1feebc0
1 parent 785679a
commit 1feebc0
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/anomalib/post_processing/one_class.py b/src/anomalib/post_processing/one_class.py
@@ -273,8 +273,7 @@ def _normalize(
         if preds is None:
             return None
         preds = ((preds - threshold) / (norm_max - norm_min)) + 0.5
-        preds = torch.minimum(preds, torch.tensor(1))
-        return torch.maximum(preds, torch.tensor(0))
+        return preds.clamp(min=0, max=1)
 
     @property
     def normalized_image_threshold(self) -> float: