yujunjun/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-06-26 21:24:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix arabic postprocess bug

Browse Source
This commit is contained in:
Topdu 2022-08-25 04:33:42 +00:00
parent 1a5e1be1fa
commit 6cded42b10
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
ppocr/postprocess/rec_postprocess.py
View File

@ -38,6 +38,10 @@ class BaseRecLabelDecode(object):
if use_space_char: if use_space_char:
self.character_str.append(" ") self.character_str.append(" ")
dict_character = list(self.character_str) 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) dict_character = self.add_special_char(dict_character)
self.dict = {} self.dict = {}
@ -45,11 +49,6 @@ class BaseRecLabelDecode(object):
self.dict[char] = i self.dict[char] = i
self.character = dict_character self.character = dict_character
if 'arabic' in character_dict_path:
self.reverse = True
else:
self.reverse = False
def pred_reverse(self, pred): def pred_reverse(self, pred):
pred_re = [] pred_re = []
c_current = '' c_current = ''