code: def load_image_test(datapoint):
      input_image = tf.image.resize(datapoint["image"], (128, 128))
      input_mask = tf.image.resize(
          datapoint["segmentation_mask"], (128, 128)
      )
      input_image, input_mask = normalize(input_image, input_mask)
      return input_image, input_mask
  
file_path: TensorFlowOnSpark/examples/segmentation/segmentation.py
imports: from __future__ import absolute_import, division, print_function, unicode_literals
  from tensorflow_examples.models.pix2pix import pix2pix
  import tensorflow_datasets
  import tensorflow
label_desc: Use image node_id 3 for calling a global function with example usage: load_image_test(datapoint) and returns: input_image,  input_mask