TensorFlow:我如何获得 tf.contrib.metrics.streaming_mean_iou 中的 total_cm [英] TensorFlow: How Can I get the total_cm in tf.contrib.metrics.streaming_mean_iou

问题描述

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

I am currently using tf.contrib.metrics.streaming_mean_iou to compute the IoU precision. This function returns only mean_iou while I want precision for each class.

我该怎么办?

推荐答案

tf.contrib.metrics.confusion_matrix 看起来像你需要的.一旦你有了混淆矩阵，你就可以 用它来计算每个类的精度.

tf.contrib.metrics.confusion_matrix looks like what you need. Once you have the confusion matrix, you can use that to compute the precision for each class.

这篇关于TensorFlow:我如何获得 tf.contrib.metrics.streaming_mean_iou 中的 total_cm的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！