import tensorflow as tf

      argmax_results = tf.image.resize_nearest_neighbor(
          tf.expand_dims(argmax_results, 3),
          tf.shape(images)[1:3],
          align_corners=True,
          name='resize_prediction')
      predictions[output] = tf.squeeze(argmax_results, 3)
      #predictions[output + PROB_SUFFIX] = tf.image.resize_bilinear(
      #    tf.nn.softmax(logits),
      #    tf.shape(images)[1:3],
      #    align_corners=True,
