torch_em.loss

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