fix encode for srn

3d8b42dc · tink2123 · 4b845ca1 · 3d8b42dc
Commit 3d8b42dc authored Feb 08, 2021 by tink2123
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

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

No files found.
--- 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