Source code for locpix.img_processing.training.loss

"""Defie loss functions"""

import torch


[docs] class dice_loss: def __call__(self, logits, target): prob = torch.sigmoid(logits) int = torch.sum(prob * target) union = torch.sum(prob) + torch.sum(target) return 1 - (2.0 * int) / (union)