Skip to content

Commit d24860e

Browse files
committed
minor fix
1 parent 6f20873 commit d24860e

1 file changed

Lines changed: 3 additions & 6 deletions

File tree

challenge/evaluation.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -87,9 +87,6 @@ def match_result(self, answer, GT):
8787
answer_nums = np.array([list(map(float, x.split()))[0] for x in answer_nums]).reshape(-1, 2)
8888
GT_nums = np.array([list(map(float, x.split()))[0] for x in GT_nums]).reshape(-1, 2)
8989

90-
if len(answer_nums) == 0:
91-
return [], 0
92-
9390
matched_out = []
9491
true_positives = 0
9592
false_positives = 0
@@ -111,9 +108,9 @@ def match_result(self, answer, GT):
111108
false_positives += 1
112109

113110
false_negatives = len(GT_nums) - true_positives
114-
precision = true_positives / (true_positives + false_positives)
115-
recall = true_positives / (true_positives + false_negatives)
116-
F1 = 2 * precision * recall / (precision + recall)
111+
precision = true_positives / (true_positives + false_positives + 1e-8)
112+
recall = true_positives / (true_positives + false_negatives + 1e-8)
113+
F1 = 2 * precision * recall / (precision + recall + 1e-8)
117114

118115
return matched_out, F1
119116

0 commit comments

Comments
 (0)