mIoU

ignite.metrics.mIoU(cm, ignore_index=None)

Calculates mean Intersection over Union using ConfusionMatrix metric.

Parameters

• cm (ConfusionMatrix) – instance of confusion matrix metric

• ignore_index (Optional[int]) – index to ignore, e.g. background index

Returns

MetricsLambda

Return type

MetricsLambda

Examples

train_evaluator = ...

cm = ConfusionMatrix(num_classes=num_classes)
mIoU(cm, ignore_index=0).attach(train_evaluator, 'mean IoU')

state = train_evaluator.run(train_dataset)
# state.metrics['mean IoU'] -> scalar