Skip to content

Commit 831e9fd

Browse files
committed
Division by 0 fixed
1 parent 3e1e060 commit 831e9fd

File tree

1 file changed

+12
-3
lines changed

1 file changed

+12
-3
lines changed

algorithms/objectDetectionMetrics.py

+12-3
Original file line numberDiff line numberDiff line change
@@ -100,9 +100,18 @@ def objects_detection_metrics(inputs):
100100
for obj in objects_predicted[i]:
101101
if not object_is_in_list(obj, image):
102102
fp += 1
103-
precision = tp / (tp + fp)
104-
recall = tp / (tp + fn)
105-
f1 = 2 * (precision * recall) / (precision + recall)
103+
if tp + fp == 0:
104+
precision = 0
105+
else:
106+
precision = tp / (tp + fp)
107+
if tp + fn == 0:
108+
recall = 0
109+
else:
110+
recall = tp / (tp + fn)
111+
if precision + recall == 0:
112+
f1 = 0
113+
else:
114+
f1 = 2 * (precision * recall) / (precision + recall)
106115
results.append({"precision": precision, "recall": recall, "f1": f1})
107116

108117
return [{"name": "results", "value": results}]

0 commit comments

Comments
 (0)