From fe694ae9023f5ed0abfc01e869f9f9445bc18893 Mon Sep 17 00:00:00 2001 From: xmy0916 <863299715@qq.com> Date: Wed, 20 Jan 2021 15:16:37 +0800 Subject: [PATCH] fix bugs --- configs/rec/multi_language/rec_multi_language_lite_train.yml | 2 +- ppocr/postprocess/rec_postprocess.py | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/configs/rec/multi_language/rec_multi_language_lite_train.yml b/configs/rec/multi_language/rec_multi_language_lite_train.yml index c42a3d1a..729dedd4 100644 --- a/configs/rec/multi_language/rec_multi_language_lite_train.yml +++ b/configs/rec/multi_language/rec_multi_language_lite_train.yml @@ -17,7 +17,7 @@ Global: # for data or label process character_dict_path: # Set the language of training, if set, select the default dictionary file - character_type: + character_type: ch max_text_length: 25 infer_mode: False use_space_char: True diff --git a/ppocr/postprocess/rec_postprocess.py b/ppocr/postprocess/rec_postprocess.py index a18e101b..62036301 100644 --- a/ppocr/postprocess/rec_postprocess.py +++ b/ppocr/postprocess/rec_postprocess.py @@ -24,7 +24,9 @@ class BaseRecLabelDecode(object): character_type='ch', use_space_char=False): support_character_type = [ - 'ch', 'en', 'en_sensitive', 'french', 'german', 'japan', 'korean' + 'ch', 'en', 'en_sensitive', 'french', 'german', 'japan', 'korean', 'it', + 'xi', 'pu', 'ru', 'ar', 'ta', 'ug', 'fa', 'ur', 'rs', 'oc', 'rsc', 'bg', + 'uk', 'be', 'te', 'ka', 'chinese_cht', 'hi', 'mr', 'ne' ] assert character_type in support_character_type, "Only {} are supported now but get {}".format( support_character_type, character_type) -- GitLab