diff --git a/dan/predict/attention.py b/dan/predict/attention.py index 160dd52bf25b1ee6d0d492924efa718acbf42d04..84ecfc394936fcb7daaeeef3c5e1b9fbfe3cefc4 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))