From ced657d5df3881d4bf8e8bb229d6ce48dbbb07b0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sol=C3=A8ne=20Tarride?= <starride@teklia.com>
Date: Thu, 13 Apr 2023 12:00:38 +0000
Subject: [PATCH] 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 84ecfc39..c9182597 100644
--- a/dan/predict/attention.py
+++ b/dan/predict/attention.py
@@ -119,7 +119,7 @@ def get_predicted_polygons_with_confidence(
polygon, _ = get_polygon(
text_piece,
max_value,
- offset,
+ start_index,
weights,
threshold_method=threshold_method,
threshold_value=threshold_value,
--
GitLab