DiceLoss
- class easypl.losses.segmentation.diceloss.DiceLoss(weight: Optional[Tensor] = None, ignore_index: Optional[int] = None, **kwargs)
Dice loss, need one hot encode input. Taken from: https://github.com/hubutui/DiceLoss-PyTorch/blob/master/loss.py. Inputs must be a tensor of shape [N, C, *].
- weight
An array of shape [num_classes,].
- Type:
Optional[torch.Tensor]
- ignore_index
Class index to ignore.
- Type:
Optional[int]
- kwargs
Additional arguments.
- Returns:
Loss value
- Return type:
torch.Tensor