torch_em.loss
Loss functions for training neural networks with PyTorch.
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"""