Skip to content
Snippets Groups Projects
Commit 66f0f92d authored by Solene Tarride's avatar Solene Tarride Committed by Yoann Schneider
Browse files

Re-fix offset in get_predicted_polygons_with_confidence()

parent ced657d5
No related branches found
No related tags found
1 merge request!101Re-fix offset in get_predicted_polygons_with_confidence()
Hide whitespace changes
Inline Side-by-side
dan/predict/attention.py
+ 1
− 1
View file @ 66f0f92d
...@@ -115,7 +115,6 @@ def get_predicted_polygons_with_confidence( ...@@ -115,7 +115,6 @@ def get_predicted_polygons_with_confidence(
polygons = [] polygons = []
start_index = 0 start_index = 0
for text_piece, confidence in zip(text_list, confidence_list): for text_piece, confidence in zip(text_list, confidence_list):
start_index += len(text_piece) + offset
polygon, _ = get_polygon( polygon, _ = get_polygon(
text_piece, text_piece,
max_value, max_value,
...@@ -125,6 +124,7 @@ def get_predicted_polygons_with_confidence( ...@@ -125,6 +124,7 @@ def get_predicted_polygons_with_confidence(
threshold_value=threshold_value, threshold_value=threshold_value,
size=(width, height), size=(width, height),
) )
start_index += len(text_piece) + offset
polygon["text"] = text_piece polygon["text"] = text_piece
polygon["text_confidence"] = confidence polygon["text_confidence"] = confidence
polygons.append(polygon) polygons.append(polygon)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment