add multi-lang

5 years ago · f3de116010
parent 28c1f2497f
commit f3de116010
2 changed files with 2 additions and 2 deletions
--- a/ppocr/data/imaug/label_ops.py
+++ b/ppocr/data/imaug/label_ops.py
@ -95,7 +95,7 @@ class BaseRecLabelEncode(object):
        support_character_type = [
            'ch', 'en', 'EN_symbol', 'french', 'german', 'japan', 'korean',
            'EN', 'it', 'es', 'pt', 'ru', 'ar', 'ta', 'ug', 'fa', 'ur',
-            'rs_latin', 'oc', 'rs_cyrillic', 'bg', 'uk', 'be', 'te', 'ka',
+            'rs_latin', 'oc', 'rs_cyrillic', 'bg', 'uk', 'be', 'te', 'kn',
            'ch_tra', 'hi', 'mr', 'ne'
        ]
        assert character_type in support_character_type, "Only {} are supported now but get {}".format(
--- a/ppocr/postprocess/rec_postprocess.py
+++ b/ppocr/postprocess/rec_postprocess.py
@ -27,7 +27,7 @@ class BaseRecLabelDecode(object):
        support_character_type = [
            'ch', 'en', 'EN_symbol', 'french', 'german', 'japan', 'korean',
            'it', 'es', 'pt', 'ru', 'ar', 'ta', 'ug', 'fa', 'ur', 'rs_latin',
-            'oc', 'rs_cyrillic', 'bg', 'uk', 'be', 'te', 'ka', 'ch_tra', 'hi',
+            'oc', 'rs_cyrillic', 'bg', 'uk', 'be', 'te', 'kn', 'ch_tra', 'hi',
            'mr', 'ne', 'EN'
        ]
        assert character_type in support_character_type, "Only {} are supported now but get {}".format(