如何通过python实现IOU计算代码实例


Posted in Python onNovember 02, 2020

Intersection over Union(IOU)是一种测量在特定数据集中检测相应物体准确度的一个标准。IoU是一个简单的测量标准,只要是在输出中得出一个预测范围(bounding boxes)的任务都可以用IoU来进行测量。

IoU分数是对象类别分割问题的标准性能度量 [1] 。 给定一组图像,IoU测量给出了在该组图像中存在的对象的预测区域和地面实况区域之间的相似性

计算两个矩形的交并比,通常在检测任务里面可以作为一个检测指标。你的预测bbox和groundtruth之间的差异,就可以通过IOU来体现。

代码如下

#!/usr/bin/env python
# encoding: utf-8
 
import numpy as np
 
'''
函数说明:计算两个框的重叠面积
输入:
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("没有重合区域")
    return 0
  else:
  #print("有重合区域")
    intersect = (right_line - left_line) * (bottom_line - top_line)
    iouv=(float(intersect) / float(sum_area - intersect))*1.0
 
    return iouv
 
'''
函数说明:获取两组匹配结果
输入:
rectA 车位
rectB 车辆
threod 重叠面积最小数值界限 默认0.6
输出:
CarUse 一维数组保存是否占用 1 占用 0 没有
 
'''
def TestCarUse(rectA,rectB,threod=0.6,debug=0):
  #threod=0.8#设定最小值
  ALength=len(rectA)
  BLength=len(rectB)
 
  #创建保存匹配结果的矩阵
  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 #有一个超过匹配认为车位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)

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
跟老齐学Python之用Python计算
Sep 12 Python
python连接oracle数据库实例
Oct 17 Python
Python判断字符串与大小写转换
Jun 08 Python
各个系统下的Python解释器相关安装方法
Oct 12 Python
高效测试用例组织算法pairwise之Python实现方法
Jul 19 Python
selenium+python实现自动登录脚本
Apr 22 Python
python 实现selenium断言和验证的方法
Feb 13 Python
python验证码图片处理(二值化)
Nov 01 Python
Python网络爬虫信息提取mooc代码实例
Mar 06 Python
更新升级python和pip版本后不生效的问题解决
Apr 17 Python
python实现Oracle查询分组的方法示例
Apr 30 Python
python中@property的作用和getter setter的解释
Dec 22 Python
Opencv常见图像格式Data Type及代码实例
Nov 02 #Python
Jupyter安装链接aconda实现过程图解
Nov 02 #Python
详解Python3.8+PyQt5+pyqt5-tools+Pycharm配置详细教程
Nov 02 #Python
python定时截屏实现
Nov 02 #Python
PyQT5速成教程之Qt Designer介绍与入门
Nov 02 #Python
如何基于matlab相机标定导出xml文件
Nov 02 #Python
PyCharm安装PyQt5及其工具(Qt Designer、PyUIC、PyRcc)的步骤详解
Nov 02 #Python
You might like
PHP实现无限极分类图文教程
2014/11/25 PHP
php校验表单检测字段是否为空的方法
2015/03/20 PHP
用PHP做了一个领取优惠券活动的示例代码
2019/07/05 PHP
Laravel 关联模型-关联新增和关联更新的方法
2019/10/10 PHP
jquery实现动态菜单的实例代码
2013/11/28 Javascript
JavaScript中的undefined学习总结
2013/11/30 Javascript
Javascript核心读书有感之词法结构
2015/02/01 Javascript
js实现百度联盟中一款不错的图片切换效果完整实例
2015/03/04 Javascript
JavaScript跨平台的开源框架NativeScript
2015/03/24 Javascript
jQuery实现类似淘宝网图片放大效果的方法
2015/07/08 Javascript
JS中的二叉树遍历详解
2016/03/18 Javascript
JS模态窗口返回值兼容问题的完美解决方法
2016/05/28 Javascript
jQuery使用经验小技巧(推荐)
2016/05/31 Javascript
仿百度换肤功能的简单实例代码
2016/07/11 Javascript
Google 地图控件集详解及实例代码
2016/08/06 Javascript
老生常谈javascript变量的命名规范和注释
2016/09/29 Javascript
微信小程序 生命周期函数详解
2017/05/24 Javascript
express+mockjs实现模拟后台数据发送功能
2018/01/07 Javascript
vue左右侧联动滚动的实现代码
2018/06/06 Javascript
11个教程中不常被提及的JavaScript小技巧(推荐)
2019/04/17 Javascript
elementUI select组件使用及注意事项详解
2019/05/29 Javascript
继承行为在 ES5 与 ES6 中的区别详解
2019/12/24 Javascript
原生JS实现多条件筛选
2020/08/19 Javascript
[30:37]【全国守擂赛】第三周擂主赛 Dark Knight vs. Leopard Gaming
2020/05/04 DOTA
python文件特定行插入和替换实例详解
2017/07/12 Python
Python学习_几种存取xls/xlsx文件的方法总结
2018/05/03 Python
python让列表倒序输出的实例
2018/06/25 Python
Numpy 改变数组维度的几种方法小结
2018/08/02 Python
Django 创建/删除用户的示例代码
2019/07/24 Python
Python对接 xray 和微信实现自动告警
2019/09/17 Python
基于python爬取有道翻译过程图解
2020/03/31 Python
python/golang实现循环链表的示例代码
2020/09/14 Python
CSS3 box-sizing属性
2009/04/17 HTML / CSS
2015年团支部年度工作总结
2015/05/27 职场文书
《LOL》“克隆大作战”久违归来 幻灵战队皮肤上线
2022/04/03 其他游戏
Golang 入门 之url 包
2022/05/04 Golang