torch_em.loss
1from .affinity_side_loss import AffinitySideLoss 2from .combined_loss import CombinedLoss 3from .contrastive import ContrastiveLoss 4from .dice import DiceLoss, dice_score 5from .spoco_loss import SPOCOLoss 6from .wrapper import ApplyAndRemoveMask, ApplyMask, LossWrapper, MaskIgnoreLabel 7from .distance_based import DistanceLoss, DiceBasedDistanceLoss 8 9EMBEDDING_LOSSES = ( 10 ContrastiveLoss, SPOCOLoss 11)
EMBEDDING_LOSSES =
(<class 'torch_em.loss.contrastive.ContrastiveLoss'>, <class 'torch_em.loss.spoco_loss.SPOCOLoss'>)