From 38fc1fae63d399a86573fc02114f8ee19ccf1826 Mon Sep 17 00:00:00 2001 From: WenmuZhou Date: Tue, 27 Apr 2021 12:12:19 +0800 Subject: [PATCH] add max_text_length to SRNLabelDecode --- ppocr/postprocess/rec_postprocess.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ppocr/postprocess/rec_postprocess.py b/ppocr/postprocess/rec_postprocess.py index e6bc07aa..d353391c 100644 --- a/ppocr/postprocess/rec_postprocess.py +++ b/ppocr/postprocess/rec_postprocess.py @@ -218,6 +218,7 @@ class SRNLabelDecode(BaseRecLabelDecode): **kwargs): super(SRNLabelDecode, self).__init__(character_dict_path, character_type, use_space_char) + self.max_text_length = kwargs.get('max_text_length', 25) def __call__(self, preds, label=None, *args, **kwargs): pred = preds['predict'] @@ -229,9 +230,9 @@ class SRNLabelDecode(BaseRecLabelDecode): preds_idx = np.argmax(pred, axis=1) preds_prob = np.max(pred, axis=1) - preds_idx = np.reshape(preds_idx, [-1, 25]) + preds_idx = np.reshape(preds_idx, [-1, self.max_text_length]) - preds_prob = np.reshape(preds_prob, [-1, 25]) + preds_prob = np.reshape(preds_prob, [-1, self.max_text_length]) text = self.decode(preds_idx, preds_prob) -- GitLab