
    3j7                    x   S r SSKJr  SSKJrJr  SSKJrJr  SSK	J
r
Jr  SSKJrJr  SSKJrJr  SS	KJrJrJrJr  SS
KJrJr  SSKJrJr  SSKJrJr  SSK J!r!J"r"  SSK#J$r$  SSK%J&r&J'r'J(r(J)r)J*r*J+r+J,r,J-r-J.r.J/r/J0r0J1r1J2r2  SSK3J3r3  SSK4J5r5J6r6  SSK7J8r8J9r9  SSK:J;r;J<r<  SSK=J>r>J=r=  SSK?J@r@JArA  SSKBJCrCJDrD  / SQrEg)zLosses submodule for Kornia.

This package provides a collection of loss functions for computer vision tasks, including SSIM,
PSNR, Dice, Focal, Lovasz, and more.
    )annotations   )
CauchyLosscauchy_loss)CharbonnierLosscharbonnier_loss)InverseDepthSmoothnessLossinverse_depth_smoothness_loss)DiceLoss	dice_loss)js_div_loss_2dkl_div_loss_2d)BinaryFocalLossWithLogits	FocalLossbinary_focal_loss_with_logits
focal_loss)GemanMcclureLossgeman_mcclure_loss)HausdorffERLossHausdorffERLoss3D)LovaszHingeLosslovasz_hinge_loss)LovaszSoftmaxLosslovasz_softmax_loss)MS_SSIMLoss)MIKernelMILossFromRefMILossFromRef2DMILossFromRef3DNMILossFromRefNMILossFromRef2DNMILossFromRef3Dmutual_information_lossmutual_information_loss_2dmutual_information_loss_3d"normalized_mutual_information_loss%normalized_mutual_information_loss_2d%normalized_mutual_information_loss_3d)one_hot)PSNRLoss	psnr_loss)SSIMLoss	ssim_loss)
SSIM3DLossssim3d_loss)TotalVariationtotal_variation)TverskyLosstversky_loss)
WelschLosswelsch_loss)1r   r   r   r   r   r   r   r   r	   r   r   r   r   r   r   r   r    r!   r"   r*   r.   r,   r0   r2   r4   r   r   r   r   r   r   r
   r   r   r   r   r#   r$   r%   r&   r'   r(   r)   r+   r/   r-   r1   r3   r5   N)F__doc__
__future__r   cauchyr   r   charbonnierr   r   depth_smoothr	   r
   dicer   r   
divergencer   r   focalr   r   r   r   geman_mcclurer   r   	hausdorffr   r   lovasz_hinger   r   lovasz_softmaxr   r   ms_ssimr   mutual_informationr   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   psnrr*   r+   ssimr,   r-   ssim3dr.   r/   r1   r0   tverskyr2   r3   welschr4   r5   __all__     P/home/wildlama/miniconda3/lib/python3.13/site-packages/kornia/losses/__init__.py<module>rM      se   $ # + : S % 6 b b ? 9 < B       % % + < . +2rK   