tf_mtl_data_loading.py

# This snippet is adapted from here: https://www.tensorflow.org/guide/datasets
def input_fn(dataframe, is_eval=False):

  # Load the list of files
  filenames = tf.constant(dataframe.iloc[:, 0].tolist())

  # Load the labels
  labels = tf.constant(dataframe.iloc[:, 1:].values.astype(np.float32))

  # Build the dataset with image processing on top of it
  dataset = tf.data.Dataset.from_tensor_slices((filenames, labels))
  dataset = dataset.map(_parse_function)

  # Add shuffling and repeatition if training
  if is_eval:
    dataset = dataset.batch(64)
  else:
    dataset = dataset.repeat().shuffle(1000).batch(64)
  
  return dataset