python不使用for計(jì)算兩組、多個(gè)矩形兩兩間的iou方式
解決問題: 不使用for計(jì)算兩組、多個(gè)矩形兩兩間的iou
使用numpy廣播的方法,在python程序中并不建議使用for語句,python中的for語句耗時(shí)較多,如果使用numpy廣播的思想將會(huì)提速不少。
代碼:
def calc_iou(bbox1, bbox2): if not isinstance(bbox1, np.ndarray): bbox1 = np.array(bbox1) if not isinstance(bbox2, np.ndarray): bbox2 = np.array(bbox2) xmin1, ymin1, xmax1, ymax1, = np.split(bbox1, 4, axis=-1) xmin2, ymin2, xmax2, ymax2, = np.split(bbox2, 4, axis=-1) area1 = (xmax1 - xmin1) * (ymax1 - ymin1) area2 = (xmax2 - xmin2) * (ymax2 - ymin2) ymin = np.maximum(ymin1, np.squeeze(ymin2, axis=-1)) xmin = np.maximum(xmin1, np.squeeze(xmin2, axis=-1)) ymax = np.minimum(ymax1, np.squeeze(ymax2, axis=-1)) xmax = np.minimum(xmax1, np.squeeze(xmax2, axis=-1)) h = np.maximum(ymax - ymin, 0) w = np.maximum(xmax - xmin, 0) intersect = h * w union = area1 + np.squeeze(area2, axis=-1) - intersect return intersect / union
程序中輸入為多個(gè)矩形[xmin, ymin, xmax,ymax]格式的數(shù)組或者list,輸出為numpy格式,例:輸入的shape為(3, 4)、(5,4)則輸出為(3, 5)各個(gè)位置為boxes間相互的iou值。后面會(huì)卡一個(gè)iou的閾值,然后就可以將滿足條件的索引取出。如:
def delete_bbox(bbox1, bbox2, roi_bbox1, roi_bbox2, class1, class2, idx1, idx2, iou_value): idx = np.where(iou_value > 0.4) left_idx = idx[0] right_idx = idx[1] left = roi_bbox1[left_idx] right = roi_bbox2[right_idx] xmin1, ymin1, xmax1, ymax1, = np.split(left, 4, axis=-1) xmin2, ymin2, xmax2, ymax2, = np.split(right, 4, axis=-1) left_area = (xmax1 - xmin1) * (ymax1 - ymin1) right_area = (xmax2 - xmin2) * (ymax2 - ymin2) left_idx = left_idx[np.squeeze(left_area < right_area, axis=-1)]#小的被刪 right_idx = right_idx[np.squeeze(left_area > right_area, axis=-1)] bbox1 = np.delete(bbox1, idx1[left_idx], 0) class1 = np.delete(class1, idx1[left_idx]) bbox2 = np.delete(bbox2, idx2[right_idx], 0) class2 = np.delete(class2, idx2[right_idx]) return bbox1, bbox2, class1, class2
IOU計(jì)算原理:
ymin = np.maximum(ymin1, np.squeeze(ymin2, axis=-1)) xmin = np.maximum(xmin1, np.squeeze(xmin2, axis=-1)) ymax = np.minimum(ymax1, np.squeeze(ymax2, axis=-1)) xmax = np.minimum(xmax1, np.squeeze(xmax2, axis=-1)) h = np.maximum(ymax - ymin, 0) w = np.maximum(xmax - xmin, 0) intersect = h * w
計(jì)算矩形間min的最大值,max的最小值,如果ymax-ymin值大于0則如左圖所示,如果小于0則如右圖所示
以上這篇python不使用for計(jì)算兩組、多個(gè)矩形兩兩間的iou方式就是小編分享給大家的全部內(nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
- 如何通過python實(shí)現(xiàn)IOU計(jì)算代碼實(shí)例
- python實(shí)現(xiàn)交并比IOU教程
- python shapely.geometry.polygon任意兩個(gè)四邊形的IOU計(jì)算實(shí)例
- python實(shí)現(xiàn)IOU計(jì)算案例
- python:目標(biāo)檢測模型預(yù)測準(zhǔn)確度計(jì)算方式(基于IoU)
- python實(shí)現(xiàn)的Iou與Giou代碼
- 淺談Python3實(shí)現(xiàn)兩個(gè)矩形的交并比(IoU)
- python計(jì)算二維矩形IOU實(shí)例
- Python計(jì)算機(jī)視覺里的IOU計(jì)算實(shí)例
- 解析目標(biāo)檢測之IoU
相關(guān)文章
Python使用matplotlib填充圖形指定區(qū)域代碼示例
這篇文章主要介紹了Python使用matplotlib填充圖形指定區(qū)域代碼示例,具有一定借鑒價(jià)值,需要的朋友可以參考下2018-01-01python數(shù)據(jù)庫操作指南之PyMysql使用詳解
PyMySQL是在Python3.x版本中用于連接MySQL服務(wù)器的一個(gè)庫,Python2 中則使用mysqldb,下面這篇文章主要給大家介紹了關(guān)于python數(shù)據(jù)庫操作指南之PyMysql使用的相關(guān)資料,需要的朋友可以參考下2023-03-03python3多重排序處理多數(shù)據(jù)的示例詳解
Python3的多重排序通常指的是對數(shù)據(jù)集合按照兩個(gè)或多個(gè)人數(shù)屬性進(jìn)行排序的過程,這可以通過將多個(gè)排序關(guān)鍵字作為元組傳遞給內(nèi)置的sorted()函數(shù)或者是使用列表推導(dǎo)式結(jié)合lambda函數(shù)完成,本文詳細(xì)分析了python3多重排序處理多數(shù)據(jù),需要的朋友可以參考下2024-07-07Python基本結(jié)構(gòu)之判斷語句的用法詳解
在程序的設(shè)計(jì)當(dāng)中,代碼并不是逐步按照順序進(jìn)行執(zhí)行的,在運(yùn)行到某一行代碼當(dāng)中,需要停下進(jìn)行判斷接下來將要運(yùn)行到那一個(gè)分支代碼,這種判斷就代表的是分支結(jié)構(gòu)。分支結(jié)構(gòu)是可以使用?if?語句來進(jìn)行判斷的,而我們本篇博客講的也是?if?語句,需要的可以了解一下2022-07-07Python?OpenCV中的drawMatches()關(guān)鍵匹配繪制方法
這篇文章主要介紹了Python?OpenCV中的drawMatches()關(guān)鍵匹配繪制方法,文章圍繞主題展開詳細(xì)的內(nèi)容介紹,具有一定的參考價(jià)值,需要的朋友可以參考一下2022-07-07Python簡單計(jì)算數(shù)組元素平均值的方法示例
這篇文章主要介紹了Python簡單計(jì)算數(shù)組元素平均值的方法,涉及Python簡單數(shù)組遍歷與數(shù)學(xué)運(yùn)算相關(guān)操作技巧,需要的朋友可以參考下2017-12-12python利用datetime模塊計(jì)算時(shí)間差
python中通過datetime模塊可以很方便的計(jì)算兩個(gè)時(shí)間的差,datetime的時(shí)間差單位可以是天、小時(shí)、秒,甚至是微秒,下面我們就來詳細(xì)看下datetime的強(qiáng)大功能吧2015-08-08