From f3de1160100d7697cb46da38cb8da49c8323fac4 Mon Sep 17 00:00:00 2001 From: tink2123 Date: Tue, 16 Mar 2021 10:23:12 +0800 Subject: [PATCH] add multi-lang --- ppocr/data/imaug/label_ops.py | 2 +- ppocr/postprocess/rec_postprocess.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ppocr/data/imaug/label_ops.py b/ppocr/data/imaug/label_ops.py index 16659773..e7101924 100644 --- 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( diff --git a/ppocr/postprocess/rec_postprocess.py b/ppocr/postprocess/rec_postprocess.py index 6a8004e5..b0517982 100644 --- 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( -- GitLab