Fixes Issue #81: Failing when region to small

dynobo · Oct 30, 2020 · 185f2bb · 185f2bb
1 parent 0112d86
commit 185f2bb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/normcap/handlers/ocr_handler.py b/normcap/handlers/ocr_handler.py
@@ -40,6 +40,10 @@ def handle(self, request: NormcapData) -> NormcapData:
         return request
 
     def img_to_dict(self, img, lang) -> list:
+        if not img.size or img.size[0] * img.size[1] <= 10:
+            self._logger.warning("Image too small for OCR. Dimensions: %s", img.size)
+            return []
+
         img.format = "PNG"  # WORKAROUND for a pyinstaller bug on Win
 
         # 0 = Orientation and script detection (OSD) only.