MIR.image_similarity.Segmentation_OverlapΒΆ

Segmentation overlap losses for multi-class outputs.

Classes

DiceSegLoss(*args, **kwargs)

Soft multi-class Dice loss computed from logits and integer labels.

SegmentationLoss(*args, **kwargs)

Weighted sum of cross-entropy and soft Dice loss for multi-class segmentation.