如何从分割掩码中找到 IoU？

Question

我正在执行图像分割任务，并且我使用的数据集只有基本事实但没有边界框或多边形。

我有 2 个类（忽略 0 作为背景）并且输出和地面实况标签在一个数组中

预测---/---标签

0|0|0|1|2 0|0|0|1|2 0|2|1|0|0 0|2|1|0|0 0|0|1|1|1 0|0|1|1|1 0|0|0|0|1 0|0|0|0|1

我如何从这些计算 IoU？

PS：我使用 python3 和 pytorch api

Answer 1

所以我才发现 jaccard_similarity_score 被认为是 IoU。

所以解决方法很简单

from sklearn.metrics import jaccard_similarity_score jac = jaccard_similarity_score(predictions, label, Normalize = True/False)

源码链接：https : //scikit-learn.org/stable/modules/generated/sklearn.metrics.jaccard_score.html#sklearn.metrics.jaccard_score