python实现IOU计算案例_Acc图像分类是什么 - CSDN文库

195 浏览量更新于2023-03-16 评论收藏 63KB PDF 举报

身份认证购VIP最低享 7 折!

领优惠券(最高得80元）

资源详情

资源评论

资源推荐

python实现实现IOU计算案例计算案例

主要介绍了python实现IOU计算案例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

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

IOU来体现。很简单的算法实现，我也随便写了一个，嗯，很简单。

1. 使用时，请注意bbox四个数字的顺序(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)

补充知识：补充知识：基于基于Python实现的实现的IOU算法算法---最简单易懂的代码实现最简单易懂的代码实现

概念介绍：

交并比:(Intersection over Union)

如上图所示，IOU值定位为两个矩形框面积的交集和并集的比值。即:

本内容试读结束，登录后可阅读更多

下载后可阅读完整内容，剩余1页未读，立即下载

评论0

weixin_38656337

粉丝: 4
资源: 921

会员权益专享

图片转文字

全年可省5，000元立即开通

最新资源

资源上传下载、课程学习等过程中有任何疑问或建议，欢迎提出宝贵意见哦~我们会及时处理！点击此处反馈