code: def parse_arg():
      try:
          idx = sys.argv.index("--custom_data")
          custom_data = str(sys.argv[idx + 1 :][0])
      except ValueError:
          custom_data = "1,224,224,3,1000"
      return custom_data
  
imports: import numpy
  from absl import flags
  import sys
label_desc: Use image node_id 1 for calling a global function with example usage: parse_arg() and returns: custom_data