fix attn encode

4b845ca1 · tink2123 · 895d44bc · 4b845ca1
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

ppocr/data/imaug/label_ops.py ppocr/data/imaug/label_ops.py +1 -1

未找到文件。
--- 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