change accuracy compute method

Chanbluky · May 21, 2019 · 91e125e · 91e125e
1 parent 5055bbf
commit 91e125e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tools/evaluate_model_utils.py b/tools/evaluate_model_utils.py
@@ -22,8 +22,8 @@ def calculate_model_precision(input_tensor, label_tensor):
     logits = tf.nn.softmax(logits=input_tensor)
     final_output = tf.expand_dims(tf.argmax(logits, axis=-1), axis=-1)
 
-    idx = tf.where(tf.equal(label_tensor, 1))
-    pix_cls_ret = tf.gather_nd(final_output, idx)
+    idx = tf.where(tf.equal(final_output, 1))
+    pix_cls_ret = tf.gather_nd(label_tensor, idx)
     accuracy = tf.count_nonzero(pix_cls_ret)
     accuracy = tf.divide(accuracy, tf.cast(tf.shape(pix_cls_ret)[0], tf.int64))