python實現(xiàn)IOU計算案例
更新時間：2020年04月12日 15:04:08 作者：木盞
這篇文章主要介紹了python實現(xiàn)IOU計算案例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
計算兩個矩形的交并比，通常在檢測任務(wù)里面可以作為一個檢測指標。你的預(yù)測bbox和groundtruth之間的差異，就可以通過IOU來體現(xiàn)。很簡單的算法實現(xiàn)，我也隨便寫了一個，嗯，很簡單。
1. 使用時，請注意bbox四個數(shù)字的順序(y0,x0,y1,x1)，順序不太一樣。
#!/usr/bin/env python
# encoding: utf-8
def compute_iou(rec1, rec2):
"""
computing IoU
:param rec1: (y0, x0, y1, x1), which reflects
(top, left, bottom, right)
:param rec2: (y0, x0, y1, x1)
:return: scala value of IoU
"""
# computing area of each rectangles
S_rec1 = (rec1[2] - rec1[0]) * (rec1[3] - rec1[1])
S_rec2 = (rec2[2] - rec2[0]) * (rec2[3] - rec2[1])
# computing the sum_area
sum_area = S_rec1 + S_rec2
# find the each edge of intersect rectangle
left_line = max(rec1[1], rec2[1])
right_line = min(rec1[3], rec2[3])
top_line = max(rec1[0], rec2[0])
bottom_line = min(rec1[2], rec2[2])
# judge if there is an intersect
if left_line >= right_line or top_line >= bottom_line:
return 0
else:
intersect = (right_line - left_line) * (bottom_line - top_line)
return (intersect / (sum_area - intersect))*1.0
if __name__=='__main__':
rect1 = (661, 27, 679, 47)
# (top, left, bottom, right)
rect2 = (662, 27, 682, 47)
iou = compute_iou(rect1, rect2)
print(iou)
補充知識：基于Python實現(xiàn)的IOU算法---最簡單易懂的代碼實現(xiàn)
概念介紹：
交并比:(Intersection over Union)
如上圖所示，IOU值定位為兩個矩形框面積的交集和并集的比值。即:
交并比的實現(xiàn)也是非常簡單的，執(zhí)行過程如下：
1. 交集形狀的寬度計算為：
IOU_W = min(x1,x2,x3,x4)+w1+w2-max(x1,x2,x3,x4)
2. 交集形狀的高度計算為：
IOU_H = min(y1,y2,y3,y4)+h1+h2-max(y1,y2,y3,y4)
其實是很簡單的幾何關(guān)系變換，上面的圖可以幫助你很好的理解這個意思。
代碼實現(xiàn)：001-IOU計算
以上這篇python實現(xiàn)IOU計算案例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了python爬蟲入門教程之糗百圖片爬蟲代碼分享,本文以抓取糗事百科內(nèi)涵圖為需求寫了一個爬蟲,,需要的朋友可以參考下2014-09-09
Python利用xmltodict模塊實現(xiàn)處理XML數(shù)據(jù)
理解和處理XML數(shù)據(jù)在Python中是一項常見任務(wù),xmltodict便是一個Python庫,用于將XML數(shù)據(jù)解析為易于處理的Python字典,下面我們就來學習一下xmltodict庫的具體使用吧2023-11-11
這篇文章主要介紹了python多線程高級鎖condition簡單用法,結(jié)合實例形式分析了condition對象常用方法及相關(guān)使用技巧,需要的朋友可以參考下2019-11-11
這篇文章主要為大家介紹了Python的模塊與包，具有一定的參考價值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來幫助2021-12-12
Python解析JSON數(shù)據(jù)的方法簡單例子
這篇文章主要給大家介紹了關(guān)于Python解析JSON數(shù)據(jù)的方法,解析JSON文件是Python中非常常見的操作,文中通過代碼介紹的非常詳細,需要的朋友可以參考下2023-09-09
Pandas缺失值填充 df.fillna()的實現(xiàn)
本文主要介紹了Pandas缺失值填充 df.fillna()的實現(xiàn)，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2022-07-07
這篇文章主要介紹了python實現(xiàn)馬丁策略的實例詳解,本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值，需要的朋友可以參考下2021-01-01
Python中的extend函數(shù)是用于將一個列表的元素添加到另一個列表中,它會將第一個列表中的元素逐個添加到第二個列表的末尾,這篇文章主要給大家介紹了關(guān)于python中extend函數(shù)舉例詳解以及對比的相關(guān)資料,需要的朋友可以參考下2024-03-03
今天小編就為大家分享一篇python numpy 按行歸一化的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-01-01
gltf,全稱是GL Transmission Format,是一種開放的3D文件格式,Python和C++是兩個非常強大的工具,下面我們就來看看如何結(jié)合這兩種語言來實現(xiàn)gltf文件的解析吧2025-03-03