mirror of
https://github.com/ail-project/ail-framework.git
synced 2024-11-23 06:37:15 +00:00
fix: [ocr] catch cv2 errors
This commit is contained in:
parent
e21257a3fe
commit
adbce24128
1 changed files with 2 additions and 1 deletions
|
@ -9,6 +9,7 @@ The OcrExtractor Module
|
||||||
##################################
|
##################################
|
||||||
# Import External packages
|
# Import External packages
|
||||||
##################################
|
##################################
|
||||||
|
import cv2
|
||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
|
@ -103,7 +104,7 @@ class OcrExtractor(AbstractModule):
|
||||||
print(image.id, languages)
|
print(image.id, languages)
|
||||||
try:
|
try:
|
||||||
texts = Ocrs.extract_text(path, languages)
|
texts = Ocrs.extract_text(path, languages)
|
||||||
except (OSError, ValueError) as e:
|
except (OSError, ValueError, cv2.error) as e:
|
||||||
self.logger.warning(e)
|
self.logger.warning(e)
|
||||||
self.obj.add_tag('infoleak:confirmed="false-positive"')
|
self.obj.add_tag('infoleak:confirmed="false-positive"')
|
||||||
texts = None
|
texts = None
|
||||||
|
|
Loading…
Reference in a new issue