CODEWITHSUNDEEP
tensorflow
LI Xuhong
LI Xuhong

Reputation: 2356

TensorFlow: How Can I get the total_cm in tf.contrib.metrics.streaming_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.

How can I do?

Upvotes: 1

Views: 1085

Answers (1)

Allen Lavoie
Allen Lavoie

Reputation: 5808

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.

Upvotes: 1

Related Questions