其他分享
首页 > 其他分享> > 【深度学习入门到精通系列】Generalized Dice loss解释

【深度学习入门到精通系列】Generalized Dice loss解释

作者:互联网

论文原文全程为:Generalized Overlap Measures for Evaluation and Validation in Medical Image Analysis 刚才分析过Dice Loss对小目标的预测是十分不利的,因为一旦小目标有部分像素预测错误,就可能会引起Dice系数大幅度波动,导致梯度变化大训练不稳定。另外从上面的代码实现可以发现,Dice Loss针对的是某一个特定类别的分割的损失。当类似于病灶分割有多个场景的时候一般都会使用多个Dice Loss,所以Generalized Dice loss就是将多个类别的Dice Loss进行整合,使用一个指标作为分割结果的量化指标。GDL Loss在类别数为2时公式如下:
在这里插入图片描述

keras代码:

def generalized_dice_coeff(y_true, y_pred):
    Ncl = y_pred.shape[-1]
    w = K.zeros(shape=(Ncl,))
 

标签:loss,Dice,Loss,shape,Ncl,Generalized,类别
来源: https://blog.51cto.com/u_14013325/2890578