diff --git a/test_tipc/readme.md b/test_tipc/readme.md index 1c637d76f9..f9e9d89e41 100644 --- a/test_tipc/readme.md +++ b/test_tipc/readme.md @@ -53,6 +53,7 @@ | SRN |rec_r50fpn_vd_none_srn | 识别 | 支持 | 多机多卡
混合精度 | - | - | | NRTR |rec_mtb_nrtr | 识别 | 支持 | 多机多卡
混合精度 | - | - | | SAR |rec_r31_sar | 识别 | 支持 | 多机多卡
混合精度 | - | - | +| SPIN |rec_r32_gaspin_bilstm_att | 识别 | 支持 | 多机多卡
混合精度 | - | - | | PGNet |rec_r34_vd_none_none_ctc_v2.0 | 端到端| 支持 | 多机多卡
混合精度 | - | - | | TableMaster |table_structure_tablemaster_train | 表格识别| 支持 | 多机多卡
混合精度 | - | - | diff --git a/tools/infer/predict_rec.py b/tools/infer/predict_rec.py index e6ba4d04ec..fdbf429be0 100755 --- 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': 'SARLabelDecode', + 'name': 'SPINLabelDecode', "character_dict_path": args.rec_char_dict_path, "use_space_char": args.use_space_char }