fix: [ocr] catch cv2 errors

ail-project · May 2, 2024 · adbce24 · adbce24
1 parent e21257a
commit adbce24
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bin/modules/OcrExtractor.py b/bin/modules/OcrExtractor.py
@@ -9,6 +9,7 @@
 ##################################
 # Import External packages
 ##################################
+import cv2
 import os
 import sys
 
@@ -103,7 +104,7 @@ def compute(self, message):
             print(image.id, languages)
             try:
                 texts = Ocrs.extract_text(path, languages)
-            except (OSError, ValueError) as e:
+            except (OSError, ValueError, cv2.error) as e:
                 self.logger.warning(e)
                 self.obj.add_tag('infoleak:confirmed="false-positive"')
                 texts = None