如何通过python实现IOU计算代码实例
作者:小蜗牛叽咕往前 发布时间:2021-07-09 02:06:44
标签:python,IOU,计算
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)
来源:https://www.cnblogs.com/kekeoutlook/p/13742624.html
0
投稿
猜你喜欢
- 本文较为详细的讲述了Python实现远程调用MetaSploit的方法,对Python的学习来说有很好的参考价值。具体实现方法如下:(1)安
- Macromedia Dreamweaver MX 2004提供了更多功能强劲的可视化设计工具、应用开
- 使用memcache来同步session是还是不错的,当然也可以通过redis来保存session,可以php开启并将Session存储到R
- Q:在给Dataloader设置worker数量(num_worker)时,到底设置多少合适?这个worker到底怎么工作的?
- 在广大网友心目中,他们就是中国互联网搜索领域的三驾马车。无论这三家搜索巨头承不承认,在网友眼中总会来将他们进行对比比较。当然,更多时候的比较
- 在单个HTML元素上利用CSS2.1实现拥有3张背景图片和2张内容图效果,或者多重边框的效果。这种渐进增强的方式适用于所有支持CSS2.1伪
- 本文实例总结了PHP session会话操作技巧。分享给大家供大家参考,具体如下:会话技术session将会话数据存储与服务器端,同时使会话
- PHP在运行时, 针对严重程度不同的错误,会给以不同的提示。 eg:在$a没声明时,直接相加,值为NULL,相加时当成0来算.但是,却提示N
- '定义变量 Dim cn,rs,Sql Sql = "sel
- 引言安装或者更新完pytorch后,运行不了,显示错误:(base) xu@xusu:~$ pythonPython 3.7.1 (defa
- 前言闲来无聊,写了一个爬虫程序获取百度疫情数据。申明一下,研究而已。而且页面应该会进程做反爬处理,可能需要调整对应xpath。Github仓
- Github 上很多大牛的代码都是Tensorflow v1 写的,比较新的文章则喜欢用Pytorch,这导致我们复现实验或者对比实验的时候
- 本文实例讲述了php数组索引与键值操作技巧。分享给大家供大家参考。具体如下:<?php $array = array("a&
- 现在需要一个写文件方法,将selenium的脚本运行结果写入test_result.log文件中首先创建写入方法def write_resu
- 看下面的一组例子:alert(true.toString());alert(false.toString());alert(1.123.to
- 数据集数据集为Barcelona某段时间内的气象数据,其中包括温度、湿度以及风速等。本文将简单搭建来对风速进行预测。特征构造对于风速的预测,
- 背景:这个库的安装不是像其他的一样的直接使用 pip install XXX的形式,而是使用原始的Git方式1、apex这是NVIDIA开发
- 还有种片面的观点认为,做网站设计与平面差不多,比如老罗发布的这则招聘中提到:年薪十万招擅长做下列网站设计风格的平面设计师一名。在专业角度,网
- 本文做的是基于opencv将视频帧转成图片输出,由于一个视频包含的帧数过多,经常我们并不是需要它的全部帧转成图片,因此我们希望可以设置每隔多
- 一、假设有这样一个原始dataframe二、提取索引(已经做了一些操作将Age为NaN的行提取出来并合并为一个dataframe,这里提取的