code: def get_data():
      ds = dataset.Mnist("train")
      ds = BatchData(ds, 128)
      return ds
  
imports: import argparse
  import numpy
  import tensorflow
  import tensorflow.contrib.slim
  from tensorpack import *
  from tensorpack.tfutils.summary import add_moving_summary
  from tensorpack.utils.gpu import change_gpu
  from embedding_data import MnistPairs, MnistTriplets, get_test_data
label_desc: Use image node_id 1 for calling the CenterModel obj's underlying member method code with example usage: obj.get_data() and returns: ds