From f34eb93e71cbc79a70aff33140fc6f109dd9c119 Mon Sep 17 00:00:00 2001 From: Jim Martens Date: Wed, 17 Jul 2019 13:56:21 +0200 Subject: [PATCH] Added missing functions to evaluate module doc Signed-off-by: Jim Martens --- src/twomartens/masterthesis/evaluate.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/twomartens/masterthesis/evaluate.py b/src/twomartens/masterthesis/evaluate.py index cb117e0..863cba4 100644 --- a/src/twomartens/masterthesis/evaluate.py +++ b/src/twomartens/masterthesis/evaluate.py @@ -20,8 +20,12 @@ Functionality to evaluate results of networks. Functions: get_number_gt_per_class(...): calculates the number of ground truth boxes per class - get_f1_score(...): computes the F1 score for every class + prepare-predictions(...): prepares the predictions for further processing match_predictions(...): matches predictions to ground truth boxes + get_precision_recall(...): computes the precision and recall values and returns them + get_f1_score(...): computes the F1 score for every class + get_mean_average_precisions(...): computes the mean average precision for each class and returns them + get_mean_average_precision(...): computes the mean average precision over all classes and returns it """ from typing import Sequence, Union, Tuple, List