polish code

2020-08-16 16:51:24 +08:00 · 2020-08-16 16:51:24 +08:00 · fe8ce9afdf
parent a3b291928b
commit fe8ce9afdf
1 changed files with 2 additions and 0 deletions
--- a/ppocr/utils/character.py
+++ b/ppocr/utils/character.py
@ -30,6 +30,8 @@ class CharacterOps(object):
            self.character_str = "0123456789abcdefghijklmnopqrstuvwxyz"
            dict_character = list(self.character_str)
        elif self.character_type == "ch":
+            if self.loss_type == "srn":
+                raise Exception("SRN can only support in character_type == en")
            character_dict_path = config['character_dict_path']
            add_space = False
            if 'use_space_char' in config: