import tensorflow as tf

      include_multiclass_scores=include_multiclass_scores,
      include_instance_masks=include_instance_masks,
      include_keypoints=include_keypoints))
  tensor_dict[fields.InputDataFields.image] = tf.squeeze(
    tensor_dict[fields.InputDataFields.image], axis=0)
  return tensor_dict
