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