fixed bug for predict_rec

2026-09-24 23:33:08 +08:00 · 2022-08-01 21:20:02 +08:00
parent b164bb0165
commit 8fa4620cce
2 changed files with 2 additions and 1 deletions
@@ -53,6 +53,7 @@
 | SRN    |rec_r50fpn_vd_none_srn         | 识别  | 支持 | 多机多卡 <br> 混合精度 | - | - |
 | NRTR   |rec_mtb_nrtr                   | 识别  | 支持 | 多机多卡 <br> 混合精度 | - | - |
 | SAR    |rec_r31_sar                    | 识别  | 支持 | 多机多卡 <br> 混合精度 | - | - |
+| SPIN   |rec_r32_gaspin_bilstm_att      | 识别  | 支持 | 多机多卡 <br> 混合精度 | - | - |
 | PGNet  |rec_r34_vd_none_none_ctc_v2.0  | 端到端| 支持 | 多机多卡 <br> 混合精度 | - | - |
 | TableMaster  |table_structure_tablemaster_train  | 表格识别| 支持 | 多机多卡 <br> 混合精度 | - | - |

@@ -83,7 +83,7 @@ class TextRecognizer(object):
            }
        elif self.rec_algorithm == "SPIN":
            postprocess_params = {
-                'name': 'SARLabelDecode',
+                'name': 'SPINLabelDecode',
                "character_dict_path": args.rec_char_dict_path,
                "use_space_char": args.use_space_char
            }