merge export_model

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
 
 

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]