correct median

mindee · Oct 5, 2023 · 434c0b5 · 434c0b5
1 parent bfe71b4
commit 434c0b5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/doctr/models/_utils.py b/doctr/models/_utils.py
@@ -72,7 +72,7 @@ def estimate_orientation(img: np.ndarray, n_ct: int = 50, ratio_threshold_for_li
         return 0  # in case no angles is found
     else:
         median = -median_low(angles)
-        return median if median != 0 else 0
+        return median if abs(median) != 0 else 0
 
 
 def rectify_crops(