import tensorflow as tf

    after applying all the transformations.
  """
  # Reshape flattened multiclass scores tensor into a 2D tensor of shape
  # [num_boxes, num_classes].
  if fields.InputDataFields.multiclass_scores in tensor_dict:
    tensor_dict[fields.InputDataFields.multiclass_scores] = tf.reshape(
      tensor_dict[fields.InputDataFields.multiclass_scores], [
        tf.shape(tensor_dict[fields.InputDataFields.groundtruth_boxes])[0],
        num_classes
      ])
  if fields.InputDataFields.groundtruth_boxes in tensor_dict:
    tensor_dict = util_ops.filter_groundtruth_with_nan_box_coordinates(
      tensor_dict)
    tensor_dict = util_ops.filter_unrecognized_classes(tensor_dict)
