code: def get_data(dataset_name):
      isTrain = dataset_name == "train"
      augmentors = fbresnet_augmentor(isTrain)
      return get_imagenet_dataflow(
          args.data, dataset_name, BATCH_SIZE, augmentors
      )
  
imports: import argparse
  import numpy
  import os
  import sys
  import cv2
  import tensorflow
  from tensorpack import *
  from tensorpack.dataflow import dataset
  from tensorpack.tfutils.sessinit import SmartInit
  from tensorpack.tfutils.summary import add_param_summary
  from tensorpack.tfutils.varreplace import remap_variables
  from tensorpack.utils.gpu import get_num_gpu
  from dorefa import get_dorefa, ternarize
  from imagenet_utils import ImageNetModel, eval_classification, fbresnet_augmentor, get_imagenet_dataflow
label_desc: Use image node_id 1 for calling a global function with example usage: get_data(dataset_name) and returns: get_imagenet_dataflow