如何从分割蒙版中找到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：我在pytorch api中使用python3

Answer 1

所以我才发现jaccard_similarity_score被视为IoU。

所以解决方案非常简单

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

源链接： http://scikit-learn.org/stable/modules/generated/sklearn.metrics.jaccard_similarity_score.html