How to calculate the mean IU score in image segmentation?

Question

How to compute the mean IU (mean Intersection over Union) score as in this paper?

Long, Jonathan, Evan Shelhamer, and Trevor Darrell. "Fully Convolutional Networks for Semantic Segmentation."

Answer 1

For each class Intersection over Union (IU) score is:

true positive / (true positive + false positive + false negative)

The mean IU is simply the average over all classes.

---

Regarding the notation in the paper:

• n_cl : the number of classes
• t_i : the total number of pixels in class i

• n_ij : the number of pixels of class i predicted to belong to class j. So for class i:

  • n_ii : the number of correctly classified pixels (true positives)
  • n_ij : the number of pixels wrongly classified (false positives)
  • n_ji : the number of pixels wrongly not classifed (false negatives)

You can find the matlab code to compute this directly in the Pascak DevKit here

Answer 2

 from sklearn.metrics import confusion_matrix    import numpy as np   def compute_iou(y_pred, y_true):      # ytrue, ypred is a flatten vector      y_pred = y_pred.flatten()      y_true = y_true.flatten()      current = confusion_matrix(y_true, y_pred, labels=[0, 1])      # compute mean iou      intersection = np.diag(current)      ground_truth_set = current.sum(axis=1)      predicted_set = current.sum(axis=0)      union = ground_truth_set + predicted_set - intersection      IoU = intersection / union.astype(np.float32)      return np.mean(IoU)