Added guardian function to catch case with zero ground truth

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
2019-07-09 13:45:01 +02:00
parent c6e7dd3c50
commit bdd0862d68

View File

@ -263,6 +263,11 @@ def get_precision_recall(number_gt_per_class: np.ndarray,
# Iterate over all classes.
for class_id in range(1, nr_classes + 1):
if number_gt_per_class[class_id] == 0:
cumulative_precisions.append([])
cumulative_recalls.append([])
continue
tp = cumulative_true_positives[class_id]
fp = cumulative_false_positives[class_id]