Skip to content

Latest commit

 

History

History
114 lines (86 loc) · 1.85 KB

losses.rst

File metadata and controls

114 lines (86 loc) · 1.85 KB
github_url:https://github.com/Project-MONAI/MONAI

Loss functions

Segmentation Losses

.. automodule:: monai.losses
.. currentmodule:: monai.losses

DiceLoss

.. autoclass:: DiceLoss
    :members:

.. autoclass:: Dice
    :members:

.. autoclass:: dice
    :members:

MaskedDiceLoss

.. autoclass:: MaskedDiceLoss
    :members:

GeneralizedDiceLoss

.. autoclass:: GeneralizedDiceLoss
    :members:

.. autoclass:: generalized_dice
    :members:

GeneralizedWassersteinDiceLoss

.. autoclass:: GeneralizedWassersteinDiceLoss
    :members:

.. autoclass:: generalized_wasserstein_dice
    :members:

DiceCELoss

.. autoclass:: DiceCELoss
    :members:

DiceFocalLoss

.. autoclass:: DiceFocalLoss
    :members:

GeneralizedDiceFocalLoss

.. autoclass:: GeneralizedDiceFocalLoss
    :members:

FocalLoss

.. autoclass:: FocalLoss
    :members:

TverskyLoss

.. autoclass:: TverskyLoss
    :members:

ContrastiveLoss

.. autoclass:: ContrastiveLoss
    :members:

Registration Losses

BendingEnergyLoss

.. autoclass:: BendingEnergyLoss
    :members:

LocalNormalizedCrossCorrelationLoss

.. autoclass:: LocalNormalizedCrossCorrelationLoss
    :members:

GlobalMutualInformationLoss

.. autoclass:: GlobalMutualInformationLoss
    :members:

Reconstruction Losses

SSIMLoss

.. autoclass:: monai.losses.ssim_loss.SSIMLoss
    :members:


Loss Wrappers

MultiScaleLoss

.. autoclass:: MultiScaleLoss
    :members:

MaskedLoss

.. autoclass:: MaskedLoss
    :members: