Skip to content
Snippets Groups Projects
Commit 35670074 authored by Solene Tarride's avatar Solene Tarride
Browse files

Fix polygon confidence score

parent 69f278a9
No related branches found
No related tags found
1 merge request!99Fix polygon confidence score
......@@ -261,7 +261,7 @@ def get_polygon(
metrics = [compute_contour_metrics(coverage_vector, cnt) for cnt in contours]
confidences, scores = map(list, zip(*metrics))
best_contour = contours[np.argmax(scores)]
confidence = round(confidences[np.argmax(scores)] / max_value, 2)
confidence = round(confidences[np.argmax(scores)], 2)
# Format for JSON
coord = polygon_to_bbx(np.squeeze(best_contour))
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment