code: def get_data(subset):
      ds = FakeData(
          [[SHAPE, SHAPE, CHANNELS], [SHAPE, SHAPE, CHANNELS]],
          1000,
          random=False,
          dtype=["uint8", "uint8"],
          domain=[(0, 255), (0, 10)],
      )
      ds = BatchData(ds, 1)
      return ds
  
imports: import argparse
  import cv2
  from tensorpack import tfv1
  from tensorpack import *
  from tensorpack.tfutils.export import ModelExporter
label_desc: Use image node_id 1 for calling a global function with example usage: get_data(subset) and returns: ds