move to pil

felixdittrich92 · Jul 8, 2024 · 73ad1ef · 73ad1ef
1 parent 29c92a1
commit 73ad1ef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxtr/models/preprocessor/base.py b/onnxtr/models/preprocessor/base.py
@@ -100,7 +100,7 @@ def __call__(self, x: Union[np.ndarray, List[np.ndarray]]) -> List[np.ndarray]:
             if (x.shape[1], x.shape[2]) != self.resize.output_size:
                 x = np.array([self.resize(sample) for sample in 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
             batches = [x]