update arabic rec model & add pred reverse function
parent
89f8b1d49f
commit
09ff249363
|
@ -12,7 +12,7 @@ Global:
|
|||
checkpoints:
|
||||
save_inference_dir:
|
||||
use_visualdl: false
|
||||
infer_img: doc/imgs_words/ch/word_1.jpg
|
||||
infer_img: ./doc/imgs_words/arabic/ar_2.jpg
|
||||
character_dict_path: ppocr/utils/dict/arabic_dict.txt
|
||||
max_text_length: &max_text_length 25
|
||||
infer_mode: false
|
||||
|
|
|
@ -45,6 +45,27 @@ 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 = ''
|
||||
for c in pred:
|
||||
if c in self.character[79:]:
|
||||
if c_current != '':
|
||||
pred_re.append(c_current)
|
||||
pred_re.append(c)
|
||||
c_current = ''
|
||||
else:
|
||||
c_current += c
|
||||
if c_current != '':
|
||||
pred_re.append(c_current)
|
||||
|
||||
return ''.join(pred_re[::-1])
|
||||
|
||||
def add_special_char(self, dict_character):
|
||||
return dict_character
|
||||
|
||||
|
@ -73,6 +94,10 @@ class BaseRecLabelDecode(object):
|
|||
conf_list = [0]
|
||||
|
||||
text = ''.join(char_list)
|
||||
|
||||
if self.reverse: # for arabic rec
|
||||
text = self.pred_reverse(text)
|
||||
|
||||
result_list.append((text, np.mean(conf_list).tolist()))
|
||||
return result_list
|
||||
|
||||
|
|
|
@ -1,4 +1,3 @@
|
|||
|
||||
!
|
||||
#
|
||||
$
|
||||
|
|
Loading…
Reference in New Issue