From f5f2f4239642f8d12c1a8bb9690502aede11e660 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Martin=20Ki=C5=A1=C5=A1?= Date: Fri, 25 Oct 2024 17:10:28 +0200 Subject: [PATCH] Add IndexError to catch expression when calculating transcription confidence -- in case when there are no logits (i.e. logits.shape[0] == 0) the confidence cannot be calculated. --- pero_ocr/document_ocr/page_parser.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pero_ocr/document_ocr/page_parser.py b/pero_ocr/document_ocr/page_parser.py index c637569..b4327ba 100644 --- a/pero_ocr/document_ocr/page_parser.py +++ b/pero_ocr/document_ocr/page_parser.py @@ -587,7 +587,7 @@ def get_line_confidence(self, line): log_probs = line.get_full_logprobs()[line.logit_coords[0]:line.logit_coords[1]] confidences = get_line_confidence(line, log_probs=log_probs) return np.quantile(confidences, .50) - except ValueError as e: + except (ValueError, IndexError) as e: logger.warning(f'PageOCR is unable to get confidence of line {line.id} due to exception: {e}.') return self.default_confidence return self.default_confidence