IoUとは
Intersection over Union; IoU
２つの領域がどれくらい重なっているかをあらわす指標。
物体検出やセグメンテーションのタスクで、正解の領域と予測の領域がどれくらい重なっているかをあらわす精度指標として使える。
A：正解領域
B：予測領域
バウンディングボックス
セグメンテーション
物体検出のバウンディングボックス
物体検出のバウンディングについてみてみる
算出
各矩形領域について
- w: 領域の幅
- h: 領域の高さ
ただし
IoUの値はどんな感じになるか
- 完全一致すると1.0。
- 全く重ならないと0.0。
- 大きめに予測しても、小さめに予測しても、分母となる
- 少しずれても大きくIoUの値が下がる。
完全に一致
| 10,000 | |
| 10,000 | |
| IoU | 1.0 |
ほぼ一致
| 9,604 | |
| 10,397 | |
| IoU | 0.924 |
一部一致
この例だと
| 5,625 | |
| 14,375 | |
| IoU | 0.391 |
この例だと
| 2,500 | |
| 17,500 | |
| IoU | 0.143 |
全く重なっていない
| 0 | |
| 20,000 | |
| IoU | 0.0 |
すごく大きく予測
| 10,000 | |
| 90,000 | |
| IoU | 0.111 |
すごく小さく予測
| 10,000 | |
| 90,000 | |
| IoU | 0.111 |
Pythonサンプル
def calc_iou(a, b):
"""
IoUを算出する。
Args:
a (list[int, int, int, int]) : 矩形Aの情報 [左上のX座標, 左上のY座標, 幅, 高さ]
b (list[int, int, int, int]) : 矩形Bの情報 [左上のX座標, 左上のY座標, 幅, 高さ]
Returns:
float : 算出したIoU
"""
x_a_min, y_a_min = a[:2]
x_a_max = x_a_min + a[2]
y_a_max = y_a_min + a[3]
x_b_min, y_b_min = b[:2]
x_b_max = x_b_min + b[2]
y_b_max = y_b_min + b[3]
dx = min(x_a_max, x_b_max) - max(x_a_min, x_b_min)
dy = min(y_a_max, y_b_max) - max(y_a_min, y_b_min)
if dx < 0 or dy < 0:
intersection = 0
else:
intersection = dx * dy
union = (a[2] * a[3]) + (b[2] * b[3]) - intersection
#print(f"dx: {dx}, dy: {dy}, intersection: {intersection}, union: {union}")
return intersection / union
Comments
Let's comment your feelings that are more than good