Merge pull request #49 from pitkley/feature/detect-orientation

Detect image orientation if the OCR supports it. Fixes #47
jonaswinkler · Feb 18, 2016 · 57bcb88 · 57bcb88
2 parents 1e7ece8 + c34d57a
commit 57bcb88
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/documents/consumer.py b/src/documents/consumer.py
@@ -28,6 +28,9 @@
 def image_to_string(args):
     self, png, lang = args
     with Image.open(os.path.join(self.SCRATCH, png)) as f:
+        if self.OCR.can_detect_orientation():
+            orientation = self.OCR.detect_orientation(f, lang=lang)
+            f = f.rotate(orientation["angle"], expand=1)
         return self.OCR.image_to_string(f, lang=lang)