diff --git a/test_tipc/readme.md b/test_tipc/readme.md index 1c637d76f99fffdfdc5a053fa0c5b9336fe4b731..f9e9d89e4198c1ad5fabdf58775c6f7b6d190322 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 e6ba4d04ec32e0971e3c44f15b800c2cd2bc6c51..fdbf429be0ef2008d05c141504fcc216987112b3 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 }