fix attn encode

4 years ago · 1599a4590f
parent e7187dac83
commit 1599a4590f
1 changed files with 1 additions and 1 deletions
--- a/ppocr/data/imaug/label_ops.py
+++ b/ppocr/data/imaug/label_ops.py
@ -215,7 +215,7 @@ class AttnLabelEncode(BaseRecLabelEncode):
            return None
        data['length'] = np.array(len(text))
        text = [0] + text + [len(self.character) - 1] + [0] * (self.max_text_len
-                                                               - len(text) - 1)
+                                                               - len(text) - 2)
        data['label'] = np.array(text)
        return data