filter_area.py 文件源码

def dice_core(y, y_pred):

    y_core = np.zeros(y.shape)
    y_core[np.where( (y==1) | (y==3) | (y==4) )[0]] = 1
    y_pred_core = np.zeros(y_pred.shape)
    y_pred_core[np.where( (y_pred==1) | (y_pred==3) | (y_pred==4) )[0] ] = 1

    return (2. * np.sum(y_pred_core*y_core) + np.finfo(float).eps) / (np.sum(y_core) + np.sum(y_pred_core)+ np.finfo(float).eps)