changed rotation function

ispras · Dec 26, 2023 · 9337554 · 9337554
1 parent 9ae8512
commit 9337554
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/dedocutils/preprocessing/skew_corrector.py b/dedocutils/preprocessing/skew_corrector.py
@@ -22,7 +22,8 @@ def preprocess(self, image: np.ndarray, parameters: Optional[dict] = None) -> Tu
         orientation_angle = parameters.get("orientation_angle", 0)
 
         if orientation_angle:
-            image = rotate_image(image, orientation_angle)
+            rotation_nums = orientation_angle // 90
+            image = np.rot90(image, rotation_nums)
 
         gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
         thresh = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV + cv2.THRESH_OTSU)[1]