TensorFlow：如何获得tf.contrib.metrics.streaming\u mean\u iou中的总cm

我目前正在使用tf.contrib.metrics.streaming_mean_iou来计算iou精度。这个函数只返回mean_iou，而我需要每个类的精度

---

我该怎么办

看起来像你需要的。一旦你有了混淆矩阵，你就可以了。

是的，它可以工作。同时，我发现这个函数的工作方式是相同的，并且有一个局部变量，它被添加到集合中。有没有办法访问局部变量？我想我明白了。我可以使用tf.local_variables（）获取所有局部变量。虽然我的程序中只有一个局部变量，但我得到了它。如果有更多的变量，我可以过滤它们的名称以获得我想要的变量。你也可以使用tf.Graph.get_tensor_by_name，我认为这应该适用于局部变量（即在你的图形对象上调用get_tensor_by_name，通常可以通过tf.get_default_Graph（）访问）。