From 8fa4620cce59ef51c6fedc322e3370d43a2f55e2 Mon Sep 17 00:00:00 2001 From: smilelite Date: Mon, 1 Aug 2022 21:20:02 +0800 Subject: [PATCH] fixed bug for predict_rec --- test_tipc/readme.md | 1 + tools/infer/predict_rec.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/test_tipc/readme.md b/test_tipc/readme.md index 1c637d76..f9e9d89e 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 e6ba4d04..fdbf429b 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 } -- GitLab