Skip to content

GitLab

Commit 4b845ca1 authored by tink2123's avatar tink2123
Browse files

fix attn encode

parent 895d44bc
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
ppocr/data/imaug/label_ops.py
View file @ 4b845ca1
...@@ -215,7 +215,7 @@ class AttnLabelEncode(BaseRecLabelEncode): ...@@ -215,7 +215,7 @@ class AttnLabelEncode(BaseRecLabelEncode):
return None return None
data['length'] = np.array(len(text)) data['length'] = np.array(len(text))
text = [0] + text + [len(self.character) - 1] + [0] * (self.max_text_len text = [0] + text + [len(self.character) - 1] + [0] * (self.max_text_len
- len(text) - 1) - len(text) - 2)
data['label'] = np.array(text) data['label'] = np.array(text)
return data return data
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment