diff --git a/bin/modules/OcrExtractor.py b/bin/modules/OcrExtractor.py index cc9ea8db..b96c030e 100755 --- a/bin/modules/OcrExtractor.py +++ b/bin/modules/OcrExtractor.py @@ -9,6 +9,7 @@ The OcrExtractor Module ################################## # Import External packages ################################## +import cv2 import os import sys @@ -103,7 +104,7 @@ class OcrExtractor(AbstractModule): 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