如何从分割掩码中找到 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