📉 Losses

Collection of popular semantic segmentation losses. Adapted from an awesome repo with pytorch utils https://github.com/BloodAxe/pytorch-toolbelt

Constants

segmentation_models_pytorch.losses.constants

segmentation_models_pytorch.losses.JaccardLoss

segmentation_models_pytorch.losses.DiceLoss

TverskyLoss ~~~~~~~~ .. autoclass:: segmentation_models_pytorch.losses.TverskyLoss

segmentation_models_pytorch.losses.FocalLoss

segmentation_models_pytorch.losses.LovaszLoss

segmentation_models_pytorch.losses.SoftBCEWithLogitsLoss

segmentation_models_pytorch.losses.SoftCrossEntropyLoss

segmentation_models_pytorch.losses.MCCLoss