Marmelator/cv-analysis-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 37 Wiki Activity

Refactor metrics

Browse Source
This commit is contained in:
Matthias Bisping 2023-01-09 15:53:53 +01:00
parent 689be75478
commit 65e9735bd9
1 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
cv_analysis/utils/test_metrics.py
View File

@ -38,16 +38,13 @@ def rectangle_from_dict(d):
def compute_page_iou(results_boxes: Iterable[Rectangle], ground_truth_boxes: Iterable[Rectangle]):
results = list(results_boxes)
truth = list(ground_truth_boxes)
if (not results) or (not truth):
return 0
iou_sum = 0
denominator = max(len(results), len(truth))
while results and truth:
gt_box = truth.pop()
def find_best_iou(gt_box):
best_match, best_iou = find_max_overlap(gt_box, results)
results.remove(best_match)
iou_sum += best_iou
score = iou_sum / denominator
return best_iou
score = sum(map(find_best_iou, truth)) / max(len(results), len(truth))
return score