From 68ffb46c336dab3434be8a1ef66ee256055fef87 Mon Sep 17 00:00:00 2001 From: topduke <784990967@qq.com> Date: Thu, 25 Aug 2022 12:39:41 +0800 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 f7763170..02bc1d14 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 = '' -- GitLab