From 66f0f92db68d23c1312c617c8556169b5423e18c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sol=C3=A8ne=20Tarride?= <starride@teklia.com>
Date: Thu, 13 Apr 2023 16:23:37 +0000
Subject: [PATCH] Re-fix offset in get_predicted_polygons_with_confidence()
---
dan/predict/attention.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/dan/predict/attention.py b/dan/predict/attention.py
index c9182597..6b858251 100644
--- a/dan/predict/attention.py
+++ b/dan/predict/attention.py
@@ -115,7 +115,6 @@ def get_predicted_polygons_with_confidence(
polygons = []
start_index = 0
for text_piece, confidence in zip(text_list, confidence_list):
- start_index += len(text_piece) + offset
polygon, _ = get_polygon(
text_piece,
max_value,
@@ -125,6 +124,7 @@ def get_predicted_polygons_with_confidence(
threshold_value=threshold_value,
size=(width, height),
)
+ start_index += len(text_piece) + offset
polygon["text"] = text_piece
polygon["text_confidence"] = confidence
polygons.append(polygon)
--
GitLab