Combined_Bce_Dice_Loss

最新推荐文章于 2024-04-10 19:06:25 发布

*Major*-莙工科技有限公司

最新推荐文章于 2024-04-10 19:06:25 发布

阅读量2.5k

点赞数 3

本文链接：https://blog.csdn.net/qq_41375318/article/details/112186737

版权

$C o m b i n e d B c e D i c e L o s s$

class SoftDiceLoss(nn.Module):
    def __init__(self, smooth=1., dims=(-2,-1)):

        super(SoftDiceLoss, self).__init__()
        self.smooth = smooth
        self.dims = dims
    
    def forward(self, x, y):

        tp = (x * y).sum(self.dims)
        fp = (x * (1 - y)).sum(self.dims)
        fn = ((1 - x) * y).sum(self.dims)
        
        dc = (2 * tp + self.smooth) / (2 * tp + fp + fn + self.smooth)
        dc = dc.mean()

        return 1 - dc
    
bce_fn = nn.BCEWithLogitsLoss()
dice_fn = SoftDiceLoss()

def Combined_Bce_Dice_Loss(y_pred, y_true):
    bce = bce_fn(y_pred, y_true)
    dice = dice_fn(y_pred.sigmoid(), y_true)
    return 0.8*bce+ 0.2*dice