File size: 851 Bytes
36c95ba |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
kornia.losses
=============
.. currentmodule:: kornia.losses
Reconstruction
--------------
.. autofunction:: ssim_loss
.. autofunction:: psnr_loss
.. autofunction:: total_variation
.. autofunction:: inverse_depth_smoothness_loss
Semantic Segmentation
---------------------
.. autofunction:: binary_focal_loss_with_logits
.. autofunction:: focal_loss
.. autofunction:: dice_loss
.. autofunction:: tversky_loss
Distributions
-------------
.. autofunction:: js_div_loss_2d
.. autofunction:: kl_div_loss_2d
Morphology
----------
.. autoclass:: HausdorffERLoss
.. autoclass:: HausdorffERLoss3D
Module
------
.. autoclass:: DiceLoss
.. autoclass:: TverskyLoss
.. autoclass:: FocalLoss
.. autoclass:: SSIMLoss
.. autoclass:: InverseDepthSmoothnessLoss
.. autoclass:: TotalVariation
.. autoclass:: PSNRLoss
.. autoclass:: BinaryFocalLossWithLogits
|