code: def main(args):
      assert (
          args.cfg is not None
      ), "No configuration file specified, please set --config"
      cfg = Config(args.cfg)
      builder = MatBuilder(cfg)
  
      paddleseg.utils.show_env_info()
      paddleseg.utils.show_cfg_info(cfg)
      os.environ["PADDLESEG_EXPORT_STAGE"] = "True"
  
      net = builder.model
      net.eval()
      if args.model_path:
          para_state_dict = paddle.load(args.model_path)
          net.set_dict(para_state_dict)
          logger.info("Loaded trained params of model successfully.")
  
      if args.input_shape is None:
          shape = [None, 3, None, None]
      else:
          shape = args.input_shape
      input_spec = get_input_spec(
          net.__class__.__name__, shape=shape, trimap=args.trimap
      )
  
      net = paddle.jit.to_static(net, input_spec=input_spec)
      if args.for_fd:
          save_name = "inference"
          yaml_name = "inference.yml"
      else:
          save_name = "model"
          yaml_name = "deploy.yaml"
      save_path = os.path.join(args.save_dir, save_name)
      paddle.jit.save(net, save_path)
  
      yml_file = os.path.join(args.save_dir, yaml_name)
      with open(yml_file, "w") as file:
          transforms = cfg.val_dataset_cfg.get(
              "transforms", [{"type": "Normalize"}]
          )
          data = {
              "Deploy": {
                  "transforms": transforms,
                  "model": save_name + ".pdmodel",
                  "params": save_name + ".pdiparams",
                  "input_shape": shape,
              },
              "ModelName": net.__class__.__name__,
          }
          yaml.dump(data, file)
  
      logger.info(f"Model is saved in {args.save_dir}.")
  
file_path: PaddleSeg/Matting/tools/export.py
imports: import argparse
  import os
  import sys
  import paddle
  import yaml
  import paddleseg
  from paddleseg.cvlibs import manager
  from paddleseg.utils import logger
  import ppmatting
  from ppmatting.utils import get_input_spec, Config, MatBuilder
label_desc: Use image node_id 2 for calling a global function with example usage: main(args) without return types