Skip to content

SparseMeanIoU

SparseMeanIoU class

dlf.metrics.sparse_mean_iou.SparseMeanIoU(num_classes, name="sparse_mean_iou", **kwargs)

Implementation of MeanIoU metric for sparse tensors

This implementation allows it to use sparse tensors as y_pred for the calculation of the mIoU. In addtion, all labels <= num_classes will be ignored.

Arguments

  • num_classes: int. Number of classes, everthing above or equal will be ignored
  • name: str. Name of the metric. Defaults to 'sparse_mean_iou'.

YAML Configuration

metrics:
    SparseMeanIoU:
        num_classes: 7