Skip to content

GitLab

Commit 1bf5367c authored by WenmuZhou's avatar WenmuZhou
Browse files

for preds, remove duplicate

parent adafa3e6
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
ppocr/postprocess/rec_postprocess.py
View file @ 1bf5367c
...@@ -107,7 +107,7 @@ class CTCLabelDecode(BaseRecLabelDecode): ...@@ -107,7 +107,7 @@ class CTCLabelDecode(BaseRecLabelDecode):
preds_idx = preds.argmax(axis=2) preds_idx = preds.argmax(axis=2)
preds_prob = preds.max(axis=2) preds_prob = preds.max(axis=2)
text = self.decode(preds_idx, preds_prob) text = self.decode(preds_idx, preds_prob, is_remove_duplicate=True)
if label is None: if label is None:
return text return text
label = self.decode(label) label = self.decode(label)
......
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