Rotate images bases on image dimensions

ReceiptManager · Nov 22, 2020 · a50fc4a · a50fc4a
1 parent 3e3d4f5
commit a50fc4a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 9 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/receipt_parser_core/enhancer.py b/receipt_parser_core/enhancer.py
@@ -79,6 +79,10 @@ def rotate_image(input_file, output_file, angle=90):
     """
     print(ORANGE + '\t~: ' + RESET + 'Rotate image' + RESET)
     with WandImage(filename=input_file) as img:
+        height, width = img.shape
+        if width < height:
+            angle = 0
+
         with img.clone() as rotated:
             rotated.rotate(angle)
             rotated.save(filename=output_file)
@@ -163,14 +167,14 @@ def detect_orientation(image):
     coords = np.column_stack(np.where(image > 0))
     angle = cv2.minAreaRect(coords)[-1]
     print(ORANGE + '\t~: ' + RESET + 'Get rotation angle:' + str(angle) + RESET)
-    return image
 
+    return image
 
 def enhance_image(img):
     img = rescale_image(img)
     img = grayscale_image(img)
     img = remove_noise(img)
-    img = detect_orientation(img)
+    # img = detect_orientation(img)
     return img