diff --git a/dan/predict/attention.py b/dan/predict/attention.py index c9182597dc18fc483cb5a81dbe069d2d1e42ce35..6b85825125577760fb746fba1779637508908dc9 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)