除了 IOU，深度学习中的分割任务还有其他好的指标吗？

Posted 2023-03-16

【中文标题】除了 IOU，深度学习中的分割任务还有其他好的指标吗？

【英文标题】：is there any other good metrics for segmentation task in deep learning apart from IOU?

【发布时间】：2021-09-28 14:53:09

【问题描述】：

想问下深度学习中的分割任务除了IOU（intersection over union）还有其他好的指标吗？

因为有时我从 IOU 得到 NaN 结果，只是想知道是否还有一些其他指标可以帮助观察模型的性能。

def IoU(
        targets: np.array,
        outputs: np.array,
) -> np.array:
    intersection = np.sum(
        outputs * targets,
        axis=(0, 1)
    )
    union = np.sum(
        outputs + targets,
        axis=(0, 1)
    ) - intersection

    return np.mean(intersection / union)

【问题讨论】：

问题是关于方法而不是代码。请看help center。此外，您的 IOU 实现中可能存在错误。

@AbhishekPrajapat 对不起，我只是重写它

【参考方案1】：

借条试试这个方法

intersection = np.logical_and(target, prediction)
union = np.logical_or(target, prediction)
iou_score = np.sum(intersection) / np.sum(union)