Merge pull request #7020 from smilelite/spin_branch

fixed predict_rec
2022-08-01 19:52:32 +08:00 · 2022-08-01 19:52:32 +08:00 · 3fc04c5b33
parent 732fc28fee b164bb0165
commit 3fc04c5b33
1 changed files with 1 additions and 1 deletions
--- a/tools/infer/predict_rec.py
+++ b/tools/infer/predict_rec.py
@ -83,7 +83,7 @@ class TextRecognizer(object):
            }
        elif self.rec_algorithm == "SPIN":
            postprocess_params = {
-                'name': 'SPINAttnLabelDecode',
+                'name': 'SARLabelDecode',
                "character_dict_path": args.rec_char_dict_path,
                "use_space_char": args.use_space_char
            }