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