From 6c1c34fb918be659dc6c61c6fd021f41b555a76f Mon Sep 17 00:00:00 2001 From: seekingdeep Date: Tue, 20 Apr 2021 05:18:56 +0300 Subject: [PATCH] increase the number of candidates for DBNet to 3000 (#98) This solves the issue when dealing with documents with a large number of words per-page. --- mmocr/models/textdet/postprocess/wrapper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmocr/models/textdet/postprocess/wrapper.py b/mmocr/models/textdet/postprocess/wrapper.py index 0a749465..5bd3ffeb 100644 --- a/mmocr/models/textdet/postprocess/wrapper.py +++ b/mmocr/models/textdet/postprocess/wrapper.py @@ -186,7 +186,7 @@ def db_decode(preds, min_text_score=0.3, min_text_width=5, unclip_ratio=1.5, - max_candidates=1000): + max_candidates=3000): """Decoding predictions of DbNet to instances. This is partially adapted from https://github.com/MhLiao/DB.