import tensorflow as tf

    images = tf.cast(input_dict[fields.InputDataFields.image], dtype=tf.float32)
    images = tf.expand_dims(images, axis=0)
    true_image_shape = tf.expand_dims(
      input_dict[fields.InputDataFields.true_image_shape], axis=0)
