From 6cded42b104c808108e94e7907fb01a06d9cdee4 Mon Sep 17 00:00:00 2001 From: Topdu <784990967@qq.com> Date: Thu, 25 Aug 2022 04:33:42 +0000 Subject: [PATCH 1/2] fix arabic postprocess bug --- ppocr/postprocess/rec_postprocess.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/ppocr/postprocess/rec_postprocess.py b/ppocr/postprocess/rec_postprocess.py index 1ebafc15f9..385cf09a9e 100644 --- a/ppocr/postprocess/rec_postprocess.py +++ b/ppocr/postprocess/rec_postprocess.py @@ -38,6 +38,10 @@ class BaseRecLabelDecode(object): if use_space_char: self.character_str.append(" ") dict_character = list(self.character_str) + if 'arabic' in character_dict_path: + self.reverse = True + else: + self.reverse = False dict_character = self.add_special_char(dict_character) self.dict = {} @@ -45,11 +49,6 @@ class BaseRecLabelDecode(object): self.dict[char] = i self.character = dict_character - if 'arabic' in character_dict_path: - self.reverse = True - else: - self.reverse = False - def pred_reverse(self, pred): pred_re = [] c_current = '' From 4f7b28e89f963423bc5359dd3c73a4efa1bff5c5 Mon Sep 17 00:00:00 2001 From: topduke <784990967@qq.com> Date: Thu, 25 Aug 2022 12:52:45 +0800 Subject: [PATCH 2/2] Update rec_postprocess.py --- ppocr/postprocess/rec_postprocess.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/ppocr/postprocess/rec_postprocess.py b/ppocr/postprocess/rec_postprocess.py index 385cf09a9e..69d3f7de9c 100644 --- a/ppocr/postprocess/rec_postprocess.py +++ b/ppocr/postprocess/rec_postprocess.py @@ -24,7 +24,7 @@ class BaseRecLabelDecode(object): def __init__(self, character_dict_path=None, use_space_char=False): self.beg_str = "sos" self.end_str = "eos" - + self.reverse = False self.character_str = [] if character_dict_path is None: self.character_str = "0123456789abcdefghijklmnopqrstuvwxyz" @@ -40,8 +40,6 @@ class BaseRecLabelDecode(object): dict_character = list(self.character_str) if 'arabic' in character_dict_path: self.reverse = True - else: - self.reverse = False dict_character = self.add_special_char(dict_character) self.dict = {}