Skip to content

Commit 6fb5646

Browse files
JasonLiTWaquariusjay
authored andcommitted
Fix multi scale prediction bug (#8000)
1 parent 7ae6bfd commit 6fb5646

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

research/deeplab/model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -162,7 +162,7 @@ def predict_labels_multi_scale(images,
162162
# Compute average prediction across different scales and flipped images.
163163
predictions = tf.reduce_mean(tf.concat(predictions, 4), axis=4)
164164
outputs_to_predictions[output] = tf.argmax(predictions, 3)
165-
predictions[output + PROB_SUFFIX] = tf.nn.softmax(predictions)
165+
outputs_to_predictions[output + PROB_SUFFIX] = tf.nn.softmax(predictions)
166166

167167
return outputs_to_predictions
168168

0 commit comments

Comments
 (0)