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