From 3e8f5ce45a92dd06e918c114a5aa0de98c4a2e5e Mon Sep 17 00:00:00 2001 From: smilelite Date: Sun, 10 Jul 2022 09:37:26 +0800 Subject: [PATCH] merge export_model --- .../rec_r32_gaspin_bilstm_att/rec_r32_gaspin_bilstm_att.yml | 1 - tools/export_model.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/test_tipc/configs/rec_r32_gaspin_bilstm_att/rec_r32_gaspin_bilstm_att.yml b/test_tipc/configs/rec_r32_gaspin_bilstm_att/rec_r32_gaspin_bilstm_att.yml index e53396a0..3999ecda 100644 --- a/test_tipc/configs/rec_r32_gaspin_bilstm_att/rec_r32_gaspin_bilstm_att.yml +++ b/test_tipc/configs/rec_r32_gaspin_bilstm_att/rec_r32_gaspin_bilstm_att.yml @@ -62,7 +62,6 @@ Loss: PostProcess: name: SPINAttnLabelDecode - character_dict_path: ./ppocr/utils/dict/spin_dict.txt use_space_char: False diff --git a/tools/export_model.py b/tools/export_model.py index 3ea0228f..15c4e35b 100755 --- a/tools/export_model.py +++ b/tools/export_model.py @@ -84,7 +84,7 @@ def export_single_model(model, arch_config, save_path, logger, quanter=None): "When there is tps in the network, variable length input is not supported, and the input size needs to be the same as during training" ) infer_shape[-1] = 100 - if arch_config["algorithm"] == "NRTR": + if arch_config["algorithm"] == "NRTR" or arch_config["algorithm"] == "SPIN": infer_shape = [1, 32, 100] elif arch_config["model_type"] == "table": infer_shape = [3, 488, 488] -- GitLab