Dice

class composer.models.loss.Dice(nclass)

Bases: torchmetrics.metric.Metric

The Dice Coefficient for evaluating image segmentation.

The Dice Coefficient measures how similar predictions and targets are. More concretely, it is computed as 2 * the Area of Overlap divided by the total number of pixels in both images.

compute()

Aggregate the state over all processes to compute the metric.

update(pred, target)

Update the state based on new predictions and targets.