回答思路
概念:IoU(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,混淆矩阵的每一行再加上每一列,最后减去对角线上的值。
核心代码: