forgetest5
/
PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

fix duplicate (#1771) 

* fix duplicate

* fix default action
This commit is contained in:
littletomatodonkey 2021-01-19 21:03:13 +08:00 committed by GitHub
parent b7e4528d3b
commit 301b0d347c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ppocr/postprocess/rec_postprocess.py
View File

@ -60,7 +60,7 @@ class BaseRecLabelDecode(object):
def add_special_char(self, dict_character):
return dict_character
def decode(self, text_index, text_prob=None, is_remove_duplicate=True):
def decode(self, text_index, text_prob=None, is_remove_duplicate=False):
""" convert text-index into text-label. """
result_list = []
ignored_tokens = self.get_ignored_tokens()
@ -110,7 +110,7 @@ class CTCLabelDecode(BaseRecLabelDecode):
text = self.decode(preds_idx, preds_prob)
if label is None:
return text
label = self.decode(label, is_remove_duplicate=False)
label = self.decode(label)
return text, label
def add_special_char(self, dict_character):