diff --git a/python/ch_ppocr_mobile_v2_rec/utils.py b/python/ch_ppocr_mobile_v2_rec/utils.py index d3b2c65..cb8c8d9 100755 --- a/python/ch_ppocr_mobile_v2_rec/utils.py +++ b/python/ch_ppocr_mobile_v2_rec/utils.py @@ -53,6 +53,7 @@ class CTCLabelDecode(object): for line in lines: line = line.decode('utf-8').strip("\n").strip("\r\n") self.character_str.append(line) + self.character_str.append(' ') dict_character = self.character_str dict_character = self.add_special_char(dict_character) @@ -64,7 +65,8 @@ class CTCLabelDecode(object): def __call__(self, preds, label=None, *args, **kwargs): preds_idx = preds.argmax(axis=2) preds_prob = preds.max(axis=2) - text = self.decode(preds_idx, preds_prob, is_remove_duplicate=True) + text = self.decode(preds_idx, preds_prob, + is_remove_duplicate=True) if label is None: return text label = self.decode(label) @@ -77,7 +79,9 @@ class CTCLabelDecode(object): def get_ignored_tokens(self): return [0] # for ctc blank - def decode(self, text_index, text_prob=None, is_remove_duplicate=False): + 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() diff --git a/python/ch_ppocr_server_v2_rec/utils.py b/python/ch_ppocr_server_v2_rec/utils.py index 0831c37..dcb6167 100755 --- a/python/ch_ppocr_server_v2_rec/utils.py +++ b/python/ch_ppocr_server_v2_rec/utils.py @@ -53,6 +53,7 @@ class CTCLabelDecode(object): for line in lines: line = line.decode('utf-8').strip("\n").strip("\r\n") self.character_str.append(line) + self.character_str.append(' ') dict_character = self.character_str dict_character = self.add_special_char(dict_character)