From 2765c3d48f4683e3238595e92095178f31e0e389 Mon Sep 17 00:00:00 2001 From: AxMM Date: Tue, 27 Jul 2021 00:46:43 +0100 Subject: [PATCH] Update misc.py Wrong calculation of FP and FN - should be a % of all image, like TP calculation. Added eps in the Precision and Recall calculation - to avoid division by zero and get inf. --- utils/misc.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/utils/misc.py b/utils/misc.py index 2a4b99b..0fd86c0 100644 --- a/utils/misc.py +++ b/utils/misc.py @@ -418,7 +418,7 @@ def write_summaries(self, was_best): def print_evaluate_results(hist, iu, epoch=0, iou_per_scale=None, - log_multiscale_tb=False): + log_multiscale_tb=False, eps=1e-8): """ If single scale: just print results for default scale @@ -455,10 +455,10 @@ def print_evaluate_results(hist, iu, epoch=0, iou_per_scale=None, total_pixels = hist.sum() class_data.append(100 * iu_TP[class_id] / total_pixels) - class_data.append(iu_FP[class_id] / iu_TP[class_id]) - class_data.append(iu_FN[class_id] / iu_TP[class_id]) - class_data.append(iu_TP[class_id] / (iu_TP[class_id] + iu_FP[class_id])) - class_data.append(iu_TP[class_id] / (iu_TP[class_id] + iu_FN[class_id])) + class_data.append(100 * iu_FP[class_id] / total_pixels) + class_data.append(100 * iu_FN[class_id] / total_pixels) + class_data.append(iu_TP[class_id] / (iu_TP[class_id] + iu_FP[class_id] + eps)) + class_data.append(iu_TP[class_id] / (iu_TP[class_id] + iu_FN[class_id] + eps)) tabulate_data.append(class_data) if log_multiscale_tb: