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 e53396a03e7064176f911bffc9401727c2c69a6b..3999ecda8bfca5d1c92daa198bdedc9cd4f9732e 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 3ea0228f857a2fadb36678ecd3b91bc865e56e46..15c4e35b3539706a8f0bd1d7fb4d99e76d484ed7 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]