code: def load_baseline_image(filename, image_shape=None):
      """Loads baseline image and makes sure it is of the right shape.
  
      Args:
        filename: file name of the image to load.
        image_shape: expected shape of the image.
  
      Returns:
        tf.Tensor with baseline image
      """
      image_path = tf.compat.v1.resource_loader.get_path_to_datafile(
          os.path.join("test_data", filename)
      )
      file = tf.io.read_file(image_path)
      baseline_image = (
          tf.cast(tf.image.decode_image(file), tf.float32) / 255.0
      )
      baseline_image = tf.expand_dims(baseline_image, axis=0)
      if image_shape is not None:
          # Graph-mode requires image shape to be known in advance.
          baseline_image = tf.ensure_shape(baseline_image, image_shape)
      return baseline_image
  
file_path: graphics/tensorflow_graphics/rendering/tests/rasterization_test_utils.py
imports: import os
  import random
  import numpy
  import tensorflow
  from tensorflow_graphics.geometry.transformation import look_at
  from tensorflow_graphics.rendering.camera import perspective
  from tensorflow_graphics.util import shape
label_desc: Use image node_id 5 for calling a global function with example usage: load_baseline_image(filename, image_shape) and returns: baseline_image