Fixed a bug where changing "max_text_length" for VisionLAN did not ch…

…ange the length of the output text (#11025)
PaddlePaddle · Oct 12, 2023 · d0d77fe · d0d77fe
1 parent 3786b27
commit d0d77fe
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tools/infer/predict_rec.py b/tools/infer/predict_rec.py
@@ -73,7 +73,8 @@ def __init__(self, args):
             postprocess_params = {
                 'name': 'VLLabelDecode',
                 "character_dict_path": args.rec_char_dict_path,
-                "use_space_char": args.use_space_char
+                "use_space_char": args.use_space_char,
+                "max_text_length": args.max_text_length
             }
         elif self.rec_algorithm == 'ViTSTR':
             postprocess_params = {