Skip to content

GitLab

Unverified Commit 301b0d34 authored by littletomatodonkey's avatar littletomatodonkey Committed by GitHub
Browse files

fix duplicate (#1771) 

* fix duplicate

* fix default action
parent b7e4528d
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
ppocr/postprocess/rec_postprocess.py
View file @ 301b0d34
...@@ -60,7 +60,7 @@ class BaseRecLabelDecode(object): ...@@ -60,7 +60,7 @@ class BaseRecLabelDecode(object):
def add_special_char(self, dict_character): def add_special_char(self, dict_character):
return 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. """ """ convert text-index into text-label. """
result_list = [] result_list = []
ignored_tokens = self.get_ignored_tokens() ignored_tokens = self.get_ignored_tokens()
...@@ -110,7 +110,7 @@ class CTCLabelDecode(BaseRecLabelDecode): ...@@ -110,7 +110,7 @@ class CTCLabelDecode(BaseRecLabelDecode):
text = self.decode(preds_idx, preds_prob) text = self.decode(preds_idx, preds_prob)
if label is None: if label is None:
return text return text
label = self.decode(label, is_remove_duplicate=False) label = self.decode(label)
return text, label return text, label
def add_special_char(self, dict_character): def add_special_char(self, dict_character):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment