fix starnet export

tools/export_model.py

@@ -47,7 +47,7 @@ def main():
         char_num = len(getattr(post_process_class, 'character'))
         config['Architecture']["Head"]['out_channels'] = char_num
     model = build_model(config['Architecture'])
-    init_model(config, model, logger)
+    # init_model(config, model, logger)
     model.eval()
 
     save_path = '{}/inference'.format(config['Global']['save_inference_dir'])
@@ -55,6 +55,10 @@ def main():
     infer_shape = [3, -1, -1]
     if config['Architecture']['model_type'] == "rec":
         infer_shape = [3, 32, -1]
+        if 'Transform' in config['Architecture'] and config['Architecture'][
+                'Transform'] is not None and config['Architecture'][
+                    'Transform']['name'] == 'TPS':
+            infer_shape[-1] = 100
 
     model = to_static(
         model,