btk.metrics.segmentation module

Implements segmentation metrics for evaluating deblending results in BTK.

class btk.metrics.segmentation.IoU(batch_size: int, *args, **kwargs)

Bases: SegmentationMetric

Intersection-over-Union class metric.

Note that this metric assumes that the input segmentation arrays are booleans. An error is raised if that condition is not met.

class btk.metrics.segmentation.SegmentationMetric(batch_size: int, *args, **kwargs)

Bases: Metric, ABC

Abstract class for segmentation metrics in BTK.