Spaces:
Sleeping
Sleeping
| 📉 Losses | |
| ========= | |
| Collection of popular semantic segmentation losses. Adapted from | |
| an awesome repo with pytorch utils https://github.com/BloodAxe/pytorch-toolbelt | |
| Constants | |
| ~~~~~~~~~ | |
| .. automodule:: segmentation_models_pytorch.losses.constants | |
| :members: | |
| JaccardLoss | |
| ~~~~~~~~~~~ | |
| .. autoclass:: segmentation_models_pytorch.losses.JaccardLoss | |
| DiceLoss | |
| ~~~~~~~~ | |
| .. autoclass:: segmentation_models_pytorch.losses.DiceLoss | |
| FocalLoss | |
| ~~~~~~~~~ | |
| .. autoclass:: segmentation_models_pytorch.losses.FocalLoss | |
| LovaszLoss | |
| ~~~~~~~~~~ | |
| .. autoclass:: segmentation_models_pytorch.losses.LovaszLoss | |
| SoftBCEWithLogitsLoss | |
| ~~~~~~~~~~~~~~~~~~~~~ | |
| .. autoclass:: segmentation_models_pytorch.losses.SoftBCEWithLogitsLoss | |
| SoftCrossEntropyLoss | |
| ~~~~~~~~~~~~~~~~~~~~ | |
| .. autoclass:: segmentation_models_pytorch.losses.SoftCrossEntropyLoss | |