From 3567007471732af941fde88e3f2866a937ab6a31 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sol=C3=A8ne=20Tarride?= <starride@teklia.com>
Date: Thu, 13 Apr 2023 08:13:53 +0200
Subject: [PATCH] Fix polygon confidence score
---
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 160dd52b..84ecfc39 100644
--- a/dan/predict/attention.py
+++ b/dan/predict/attention.py
@@ -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))
--
GitLab