图像分割评价指标代码

常用的图像分割评价指标包括 Jaccard 系数、Dice 系数、混淆矩阵等。以下是 Python 代码实现 Jaccard 系数和 Dice 系数的计算：

import numpy as np

def jaccard_index(y_true, y_pred):
    """
    计算 Jaccard 系数
    y_true: 真实标签，二维数组
    y_pred: 预测标签，二维数组
    """
    intersection = np.logical_and(y_true, y_pred).sum()
    union = np.logical_or(y_true, y_pred).sum()
    
    if union == 0:
        return 0
    
    return intersection / union

def dice_coefficient(y_true, y_pred):
    """
    计算 Dice 系数
    y_true: 真实标签，二维数组
    y_pred: 预测标签，二维数组
    """
    intersection = np.logical_and(y_true, y_pred).sum()
    total = y_true.sum() + y_pred.sum()
    
    if total == 0:
        return 0
    
    return 2. * intersection / total

其中，y_true 和 y_pred 都是二维数组，每个元素表示相应像素的标签，例如 1 表示前景，0 表示背景。