torch_em.loss

 1"""Loss functions for training neural networks with PyTorch.
 2"""
 3
 4from .affinity_side_loss import AffinitySideLoss
 5from .cldice import SoftSkeletonize, cldice_score, SoftclDiceLoss, CombinedclDiceLoss
 6from .combined_loss import CombinedLoss
 7from .contrastive import ContrastiveLoss
 8from .dice import DiceLoss, dice_score
 9from .spoco_loss import SPOCOLoss
10from .wrapper import ApplyAndRemoveMask, ApplyMask, LossWrapper, MaskIgnoreLabel
11from .distance_based import DistanceLoss, DiceBasedDistanceLoss
12
13EMBEDDING_LOSSES = (ContrastiveLoss, SPOCOLoss)
14"""@private
15"""