fix encode for srn

2021-02-08 03:21:01 +00:00 · 2021-02-08 03:21:01 +00:00 · 3d8b42dc0c
parent 4b845ca159
commit 3d8b42dc0c
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