CIRCL
/
AIL-framework
mirror of https://github.com/CIRCL/AIL-framework
2
1
Fork 0
Code Issues Releases Wiki Activity

fix: [qrcode extractor] fix catch cv2 exception

pull/607/merge
terrtia 2024-10-03 09:24:42 +02:00
parent ef932545b2
commit 970a169270
No known key found for this signature in database
GPG Key ID: 1E1B1F50D84613D0
1 changed files with 16 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
bin/modules/QrCodeReader.py
View File

@ -69,16 +69,22 @@ class QrCodeReader(AbstractModule):
if not contents: if not contents:
detector = cv2.QRCodeDetector() detector = cv2.QRCodeDetector()
try:
qr, decodeds, qarray, _ = detector.detectAndDecodeMulti(image) qr, decodeds, qarray, _ = detector.detectAndDecodeMulti(image)
if qr: if qr:
qr_codes = True qr_codes = True
for d in decodeds: for d in decodeds:
if d: if d:
contents.append(d) contents.append(d)
except cv2.error as e:
self.logger.error(f'{e}: {self.obj.get_global_id()}')
try:
data_qr, box, qrcode_image = detector.detectAndDecode(image) data_qr, box, qrcode_image = detector.detectAndDecode(image)
if data_qr: if data_qr:
contents.append(data_qr) contents.append(data_qr)
qr_codes = True qr_codes = True
except cv2.error as e:
self.logger.error(f'{e}: {self.obj.get_global_id()}')
if qr_codes and not contents: if qr_codes and not contents:
# # # # 0.5s per image # # # # 0.5s per image