def __init__(self, soft_dice_kwargs, ce_kwargs, aggregate="sum", square_dice=False): super(DC_and_topk_loss, self).__init__() self.aggregate = aggregate self.ce = TopKLoss(**ce_kwargs) if not square_dice: self.dc = SoftDiceLoss(apply_nonlin=softmax_helper, **soft_dice_kwargs) else: self.dc = SoftDiceLossSquared(apply_nonlin=softmax_helper, **soft_dice_kwargs)
def __init__(self, plans_file, fold, output_folder=None, dataset_directory=None, batch_dice=True, stage=None, unpack_data=True, deterministic=True, fp16=False): super().__init__(plans_file, fold, output_folder, dataset_directory, batch_dice, stage, unpack_data, deterministic, fp16) self.loss = TopKLoss(k=10)
def __init__(self, soft_dice_kwargs, topk_kwargs, focal_kwargs, aggregate="sum", square_dice=False): super(DC_topk_focal_loss, self).__init__() self.aggregate = aggregate self.topk = TopKLoss(**topk_kwargs) self.focal = FocalLoss(apply_nonlin=softmax_helper, **focal_kwargs) if not square_dice: self.dc = SoftDiceLoss(apply_nonlin=softmax_helper, **soft_dice_kwargs) else: self.dc = SoftDiceLossSquared(apply_nonlin=softmax_helper, **soft_dice_kwargs)
def __init__(self, soft_dice_kwargs, ce_kwargs, aggregate="sum"): super(DC_and_topk_loss, self).__init__() self.aggregate = aggregate self.ce = TopKLoss(**ce_kwargs) self.dc = SoftDiceLoss(apply_nonlin=softmax_helper, **soft_dice_kwargs)
def __init__(self, soft_iou_kwargs, ce_kwargs, aggregate="sum"): super(Iou_and_TopK_loss, self).__init__() self.aggregate = aggregate self.topk = TopKLoss(**ce_kwargs) self.iou = IoULoss(apply_nonlin=softmax_helper, **soft_iou_kwargs)