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] 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 = ''