列表

详情


67. 手写MIOU

回答思路

概念:IoU(Intersection over Union)是交并比,计算式为:

MioU(Mean Intersection over Union)是均交并比,属于语义分割的标准度量,计算两个集合的交并比,在语义分割的问题中,这两个集合为真实值和预测值,计算式如下:

等价于:

MIoU计算两圆交集(橙色部分)与两圆并集(红色+橙色+黄色)之间的比例,理想情况下两圆重合,比例为1。

流程:

(1)求出混淆矩阵,假设有150个样本数据,预测为1, 2, 3类各为50个,分类结束后得到的混淆矩阵为:

预测

预测

 

 

 

类1

类2

类3

 

 

实际

类1

43

2

0

类2

5

45

1

 

类3

2

3

49

每一行之和表示该类别的真实样本数量,每一列之和表示被预测为该类别的样本数量。

(2)求解miou,混淆矩阵的每一行再加上每一列,最后减去对角线上的值。

核心代码:



上一题