remove max check

felixdittrich92 · Jul 9, 2024 · 7391db8 · 7391db8
1 parent a2b1e9b
commit 7391db8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxtr/models/preprocessor/base.py b/onnxtr/models/preprocessor/base.py
@@ -71,7 +71,7 @@ def sample_transforms(self, x: np.ndarray) -> np.ndarray:
         # Resizing
         x = self.resize(x)
         # Data type & 255 division
-        if x.dtype == np.uint8 or np.max(x) > 1:
+        if x.dtype == np.uint8:
             x = x.astype(np.float32) / 255.0
 
         return x