code: def parse_args():
      parser = argparse.ArgumentParser(description="Train a detector")
      parser.add_argument("config", help="train config file path")
      parser.add_argument(
          "--shape",
          type=int,
          nargs="+",
          default=[640, 640],
          help="input image size",
      )
      args = parser.parse_args()
      return args
  
imports: import argparse
  import torch
  from fvcore.nn import FlopCountAnalysis, flop_count_table
  from mmengine import Config
  from mmengine.registry import init_default_scope
  from mmocr.registry import MODELS
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args() and returns: args