本篇文章給大家分享的是有關(guān)使用python如何實現(xiàn)IOU計算,小編覺得挺實用的,因此分享給大家學(xué)習(xí),希望大家閱讀完這篇文章后可以有所收獲,話不多說,跟著小編一起來看看吧。
Intersection over Union(IOU)是一種測量在特定數(shù)據(jù)集中檢測相應(yīng)物體準(zhǔn)確度的一個標(biāo)準(zhǔn)。IoU是一個簡單的測量標(biāo)準(zhǔn),只要是在輸出中得出一個預(yù)測范圍(bounding boxes)的任務(wù)都可以用IoU來進(jìn)行測量。
IoU分?jǐn)?shù)是對象類別分割問題的標(biāo)準(zhǔn)性能度量 [1] 。 給定一組圖像,IoU測量給出了在該組圖像中存在的對象的預(yù)測區(qū)域和地面實況區(qū)域之間的相似性
計算兩個矩形的交并比,通常在檢測任務(wù)里面可以作為一個檢測指標(biāo)。你的預(yù)測bbox和groundtruth之間的差異,就可以通過IOU來體現(xiàn)。
代碼如下
#!/usr/bin/env python # encoding: utf-8 import numpy as np ''' 函數(shù)說明:計算兩個框的重疊面積 輸入: rec1 第一個框xmin ymin xmax ymax rec2 第二個框xmin ymin xmax ymax 輸出: iouv 重疊比例 0 沒有 ''' def compute_iou(rec1, rec2): # computing area of each rectangles S_rec1 = (rec1[2] - rec1[0]) * (rec1[3] - rec1[1]) # H1*W1 S_rec2 = (rec2[2] - rec2[0]) * (rec2[3] - rec2[1]) # H2*W2 # computing the sum_area sum_area = S_rec1 + S_rec2 #總面積 # find the each edge of intersect rectangle left_line = max(rec1[0], rec2[0]) right_line = min(rec1[2], rec2[2]) top_line = max(rec1[1], rec2[1]) bottom_line = min(rec1[3], rec2[3]) # judge if there is an intersect if left_line >= right_line or top_line >= bottom_line: #print("沒有重合區(qū)域") return 0 else: #print("有重合區(qū)域") intersect = (right_line - left_line) * (bottom_line - top_line) iouv=(float(intersect) / float(sum_area - intersect))*1.0 return iouv ''' 函數(shù)說明:獲取兩組匹配結(jié)果 輸入: rectA 車位 rectB 車輛 threod 重疊面積最小數(shù)值界限 默認(rèn)0.6 輸出: CarUse 一維數(shù)組保存是否占用 1 占用 0 沒有 ''' def TestCarUse(rectA,rectB,threod=0.6,debug=0): #threod=0.8#設(shè)定最小值 ALength=len(rectA) BLength=len(rectB) #創(chuàng)建保存匹配結(jié)果的矩陣 recIOU=np.zeros((ALength,BLength),dtype=float,order='C') #用于記錄車位能夠使否占用 CarUse=np.zeros((1,ALength),dtype=int,order='C') for i in range(0,ALength): for j in range(0,BLength): iou = compute_iou(rectA[i], rectB[j]) recIOU[i][j]=format(iou,'.3f') if iou>=threod: CarUse[0,i]=1 #有一個超過匹配認(rèn)為車位i被占用 if debug==1: print('----匹配矩陣----') print(recIOU) ''' print('----車位占用情況----') for i in range(0,ALength): msg='車位'+str(i)+"-"+str(CarUse[0][i]) print(msg) ''' return CarUse if __name__=='__main__': #A代表車位 rectA1 = (30, 10, 70, 20) rectA2 = (70, 10, 80, 20) rectA =[rectA1,rectA2] #B代表檢測車輛 rectB1 = (20, 10, 35, 20) rectB2 = (30, 15, 70, 25) rectB3 = (70, 10, 80, 20) rectB =[rectB1,rectB2,rectB3] #獲取車位占用情況 rectA車位 rectB車輛 0.6占面積最小比 CarUse=TestCarUse(rectA,rectB,0.6,1) print('----車位占用情況----') for i in range(0,len(CarUse)+1): msg='車位'+str(i)+"-"+str(CarUse[0][i]) print(msg)
分享標(biāo)題:使用python如何實現(xiàn)IOU計算-創(chuàng)新互聯(lián)
標(biāo)題來源:http://www.rwnh.cn/article46/jdceg.html
成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián),為您提供手機(jī)網(wǎng)站建設(shè)、虛擬主機(jī)、微信公眾號、網(wǎng)站策劃、品牌網(wǎng)站設(shè)計、外貿(mào)建站
聲明:本網(wǎng)站發(fā)布的內(nèi)容(圖片、視頻和文字)以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主,如果涉及侵權(quán)請盡快告知,我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場,如需處理請聯(lián)系客服。電話:028-86922220;郵箱:631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載,或轉(zhuǎn)載時需注明來源: 創(chuàng)新互聯(lián)
猜你還喜歡下面的內(nèi)容