for preds, remove duplicate

1bf5367c · WenmuZhou · adafa3e6 · 1bf5367c
Commit 1bf5367c authored Jan 20, 2021 by WenmuZhou
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

ppocr/postprocess/rec_postprocess.py ppocr/postprocess/rec_postprocess.py +1 -1

No files found.
--- a/ppocr/postprocess/rec_postprocess.py
+++ b/ppocr/postprocess/rec_postprocess.py
@@ -107,7 +107,7 @@ class CTCLabelDecode(BaseRecLabelDecode):

        preds_idx = preds.argmax(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:
            return text
        label = self.decode(label)