Merge pull request #7065 from smilelite/spin_branch
spin复现合入:fixed bug for predict_recpull/7072/head
commit
c1f9e807fb
|
@ -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
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue