We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7ae6bfd commit 6fb5646Copy full SHA for 6fb5646
research/deeplab/model.py
@@ -162,7 +162,7 @@ def predict_labels_multi_scale(images,
162
# Compute average prediction across different scales and flipped images.
163
predictions = tf.reduce_mean(tf.concat(predictions, 4), axis=4)
164
outputs_to_predictions[output] = tf.argmax(predictions, 3)
165
- predictions[output + PROB_SUFFIX] = tf.nn.softmax(predictions)
+ outputs_to_predictions[output + PROB_SUFFIX] = tf.nn.softmax(predictions)
166
167
return outputs_to_predictions
168
0 commit comments