快捷導(dǎo)航

python實(shí)現(xiàn)IOU計(jì)算案例

更新時(shí)間：2020年04月12日 15:04:08 作者：木盞

這篇文章主要介紹了python實(shí)現(xiàn)IOU計(jì)算案例，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧

計(jì)算兩個(gè)矩形的交并比，通常在檢測(cè)任務(wù)里面可以作為一個(gè)檢測(cè)指標(biāo)。你的預(yù)測(cè)bbox和groundtruth之間的差異，就可以通過IOU來體現(xiàn)。很簡(jiǎn)單的算法實(shí)現(xiàn)，我也隨便寫了一個(gè)，嗯，很簡(jiǎn)單。

1. 使用時(shí)，請(qǐng)注意bbox四個(gè)數(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)

補(bǔ)充知識(shí)：基于Python實(shí)現(xiàn)的IOU算法---最簡(jiǎn)單易懂的代碼實(shí)現(xiàn)

概念介紹：

交并比:(Intersection over Union)

如上圖所示，IOU值定位為兩個(gè)矩形框面積的交集和并集的比值。即:

交并比的實(shí)現(xiàn)也是非常簡(jiǎn)單的，執(zhí)行過程如下：

1. 交集形狀的寬度計(jì)算為：

IOU_W = min(x1,x2,x3,x4)+w1+w2-max(x1,x2,x3,x4)

2. 交集形狀的高度計(jì)算為：

IOU_H = min(y1,y2,y3,y4)+h1+h2-max(y1,y2,y3,y4)

其實(shí)是很簡(jiǎn)單的幾何關(guān)系變換，上面的圖可以幫助你很好的理解這個(gè)意思。

代碼實(shí)現(xiàn)：001-IOU計(jì)算

以上這篇python實(shí)現(xiàn)IOU計(jì)算案例就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

相關(guān)文章

pycharm自定義TODO類注釋以及高亮顏色的設(shè)置方法
這篇文章主要介紹了pycharm自定義TODO類注釋以及高亮顏色的設(shè)置方法,文中通過圖文結(jié)合的方式給大家介紹的非常詳細(xì),具有一定的參考價(jià)值,需要的朋友可以參考下
2024-03-03
python接口自動(dòng)化之ConfigParser配置文件的使用詳解
這篇文章主要介紹了python接口自動(dòng)化之ConfigParser配置文件的使用,本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下
2020-08-08
Python 編碼規(guī)范(Google Python Style Guide)
本項(xiàng)目并非 Google 官方項(xiàng)目, 而是由國內(nèi)程序員憑熱情創(chuàng)建和維護(hù),對(duì)于想學(xué)習(xí)python的朋友可以參考一下
2018-05-05
基于Python+Turtle實(shí)現(xiàn)繪制簡(jiǎn)易的大風(fēng)車
大風(fēng)車，吱呀吱呦呦地轉(zhuǎn)，這里的風(fēng)景呀真好看！天好看，地好看……一首熟悉的歌曲，是否已經(jīng)把你拉回了童年？本文將用Turtle庫繪制簡(jiǎn)易的大風(fēng)車，需要的可以參考一下
2022-06-06
Tensorflow中TFRecord生成與讀取的實(shí)現(xiàn)
TFRecord格式的文件存儲(chǔ)形式會(huì)很合理的幫我們存儲(chǔ)數(shù)據(jù)，本文主要介紹了Tensorflow中TFRecord生成與讀取的實(shí)現(xiàn)，文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧
2022-05-05
python游戲地圖最短路徑求解
這篇文章主要為大家詳細(xì)介紹了python游戲地圖最短路徑的求解，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下
2019-01-01
通過實(shí)例解析python subprocess模塊原理及用法
這篇文章主要介紹了通過實(shí)例解析python subprocess模塊原理及用法,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下
2020-10-10
Python中將字符串轉(zhuǎn)換為列表的常用八種方法
本文介紹了Python中將字符串轉(zhuǎn)換為列表的八種常用方法,包括split()方法、列表解析、正則表達(dá)式、str()函數(shù)、map()函數(shù)、re.split()、re.finditer()和逐字符遍歷,感興趣的可以了解一下
2024-07-07
Python使用gensim計(jì)算文檔相似性
在文本處理中，比如商品評(píng)論挖掘，有時(shí)需要了解每個(gè)評(píng)論分別和商品的描述之間的相似度，以此衡量評(píng)論的客觀性。那么python 里面有計(jì)算文本相似度的程序包嗎，恭喜你，不僅有，而且很好很強(qiáng)大。下面我們就來體驗(yàn)下gensim的強(qiáng)大
2016-04-04
python中的GUI實(shí)現(xiàn)計(jì)算器
這篇文章主要介紹了如何利用python中的GUI實(shí)現(xiàn)計(jì)算器，文章教大家用用python的GUI做界面布局，計(jì)算器代碼熟悉控件的使用方法、優(yōu)化計(jì)算器代碼，解決獲取按鈕文本的方法，具有一定的參考價(jià)值,需要的朋友可以參考一下
2021-12-12