From 1599a4590fd70d8bec2db3708d1d1a607cbc3aee Mon Sep 17 00:00:00 2001 From: tink2123 Date: Mon, 8 Feb 2021 03:30:27 +0000 Subject: [PATCH] fix attn encode --- ppocr/data/imaug/label_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppocr/data/imaug/label_ops.py b/ppocr/data/imaug/label_ops.py index 6d401277..7a32d870 100644 --- 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 -- GitLab