Skip to content

GitLab

Commit 6fb5646f authored by Jason421's avatar Jason421 Committed by aquariusjay
Browse files

Fix multi scale prediction bug (#8000)

parent 7ae6bfdd
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
research/deeplab/model.py
View file @ 6fb5646f
...@@ -162,7 +162,7 @@ def predict_labels_multi_scale(images, ...@@ -162,7 +162,7 @@ def predict_labels_multi_scale(images,
# Compute average prediction across different scales and flipped images. # Compute average prediction across different scales and flipped images.
predictions = tf.reduce_mean(tf.concat(predictions, 4), axis=4) predictions = tf.reduce_mean(tf.concat(predictions, 4), axis=4)
outputs_to_predictions[output] = tf.argmax(predictions, 3) outputs_to_predictions[output] = tf.argmax(predictions, 3)
predictions[output + PROB_SUFFIX] = tf.nn.softmax(predictions) outputs_to_predictions[output + PROB_SUFFIX] = tf.nn.softmax(predictions)
return outputs_to_predictions return outputs_to_predictions
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment