TensorFlow：如何获取tf.contrib.metrics.streaming_mean_iou中的total_cm

Question

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

我该怎么办？

Answer 1

tf.contrib.metrics.confusion_matrix看起来就像你需要的那样。一旦有了混淆矩阵，就可以use that to compute the precision for each class。