TLDR-547 changed orientation correction function (#16)

alexander1999-hub · Alexander Golodkov · web-flow · commit f5d7401de1a4 · 2023-12-26T16:20:23.000+03:00
Co-authored-by: Alexander Golodkov &lt;golodkov@ispras.ru&gt;
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,6 +1,10 @@
 Changelog
 =========
 
+v0.3.6 (2023-12-26)
+-------------------
+* Fix image orientation correction in `SkewCorrector`
+
 v0.3.5 (2023-10-02)
 -------------------
 * Delete `Serializable` class
diff --git a/VERSION b/VERSION
@@ -1 +1 @@
-0.3.5
+0.3.6
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]
