code: def parse_args():
      parser = argparse.ArgumentParser(
          description="Extract MMProjector or GeoSampler weights"
      )
      parser.add_argument("--model-path", type=str, help="model folder")
      parser.add_argument("--output", type=str, help="output file")
      parser.add_argument(
          "--keys_to_match",
          type=str,
          default="region_geo_sampler",
          choices=["mm_projector", "region_geo_sampler"],
          help="keys to be matched",
      )
      args = parser.parse_args()
      return args
  
imports: import os
  import argparse
  import torch
  import json
  from collections import defaultdict
label_desc: Use image node_id 1 for calling a global function with example usage: parse_args() and returns: args