fix encode for srn

5 years ago · e7187dac83
parent a3afc162fa
commit e7187dac83
1 changed files with 1 additions and 1 deletions
--- a/ppocr/data/imaug/label_ops.py
+++ b/ppocr/data/imaug/label_ops.py
@ -261,7 +261,7 @@ class SRNLabelEncode(BaseRecLabelEncode):
        if len(text) > self.max_text_len:
            return None
        data['length'] = np.array(len(text))
-        text = text + [char_num] * (self.max_text_len - len(text))
+        text = text + [char_num - 1] * (self.max_text_len - len(text))
        data['label'] = np.array(text)
        return data