OneHotIoU

Description

Computes the Intersection-Over-Union metric for one-hot encoded labels. Type : polymorphic.

Input parameters

 y_pred : array, predicted values.
 y_true : array, true values (one hot encoding for example, [0, 0, 1] for 3-class problem).

Output parameters

one_hot_iou : float, result.

Calculation

This metric can be used to compute IoU for multi-class classification tasks where the labels are one-hot encoded (the last axis should have one dimension per class). If the boolean parameter sparse_y_pred is set to true, this means that the predictions are not encoded in one-hot, but are sparse integers. Consequently, it is not necessary to convert them to integer format using argmax on the class axis. Otherwise, if sparse_y_pred is false, then the labels and predictions are first converted to integer format using argmax on the class axis before the IoU is calculated.

So, depending on the value of sparse_y_pred, OneHotIoU can accommodate both one-hot encoded predictions and sparse integer predictions.

Note that if there is only one channel in the labels and predictions, this class is identical to the IoU class. In this case, use IoU instead.

Example

All these exemples are snippets PNG, you can drop these Snippet onto the block diagram and get the depicted code added to your VI (Do not forget to install Deep Learning library to run it).