在没有显式调用更新op的情况下计算与tensorflow op的均值相交?

时间:2018-07-16 15:16:03

标签: python tensorflow

发生这种情况:

lesson.course_session

似乎我必须显式调用update op(conf_mat),然后才能获得联合的均值交集。 有没有一种方法可以在不显式调用update op的情况下计算结果?

1 个答案:

答案 0 :(得分:1)

是的,您可以使用tf.control_depenencies强制在update_op节点之前执行miou

import tensorflow as tf
labels = tf.constant([1, 1, 1])
predictions = tf.constant([0, 0, 1])
miou, conf_mat = tf.metrics.mean_iou(labels, predictions, 2)
with tf.control_dependencies([tf.identity(conf_mat)]):
    miou = tf.identity(miou)
sess = tf.InteractiveSession()
sess.run(tf.local_variables_initializer())

print(miou.eval())