From f9228a3e6ab585f51f1d1e9ae7c8eda4cd954063 Mon Sep 17 00:00:00 2001 From: zhouyaqiang Date: Tue, 2 Feb 2021 09:38:20 +0800 Subject: [PATCH] Add rpn proposals --- model_zoo/official/cv/deeptext/src/config.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/model_zoo/official/cv/deeptext/src/config.py b/model_zoo/official/cv/deeptext/src/config.py index 441e5a45a1..854052ce35 100644 --- a/model_zoo/official/cv/deeptext/src/config.py +++ b/model_zoo/official/cv/deeptext/src/config.py @@ -86,12 +86,12 @@ config = ed({ # test proposal "rpn_nms_across_levels": False, - "rpn_nms_pre": 300, - "rpn_nms_post": 300, - "rpn_max_num": 300, + "rpn_nms_pre": 1000, + "rpn_nms_post": 1000, + "rpn_max_num": 1000, "rpn_nms_thr": 0.7, "rpn_min_bbox_min_size": 0, - "test_score_thr": 0.95, + "test_score_thr": 0.80, "test_iou_thr": 0.5, "test_max_per_img": 100, "test_batch_size": 2,