From 24dfd77a5b98c29799d082dfb7c248f2f32e71a5 Mon Sep 17 00:00:00 2001 From: zhouyaqiang Date: Wed, 20 Jan 2021 09:31:33 +0800 Subject: [PATCH] Add F1 score of deeptext's evaluation --- model_zoo/official/cv/deeptext/eval.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/model_zoo/official/cv/deeptext/eval.py b/model_zoo/official/cv/deeptext/eval.py index 7846b1288f..280782ef48 100644 --- a/model_zoo/official/cv/deeptext/eval.py +++ b/model_zoo/official/cv/deeptext/eval.py @@ -114,7 +114,9 @@ def Deeptext_eval_test(dataset_path='', ckpt_path=''): print("\n========================================\n") for i in range(config.num_classes - 1): j = i + 1 - print("class {} precision is {:.2f}%, recall is {:.2f}%".format(j, precisions[j] * 100, recalls[j] * 100)) + F1 = (2 * precisions[j] * recalls[j]) / (precisions[j] + recalls[j] + 1e-6) + print("class {} precision is {:.2f}%, recall is {:.2f}%," + "F1 is {:.2f}%".format(j, precisions[j] * 100, recalls[j] * 100, F1 * 100)) if config.use_ambigous_sample: break