两个Bounding Box的IOU计算代码

MA201506

于 2019-02-28 23:52:00 发布

阅读量1.5k

点赞数

CC 4.0 BY-SA版权

文章标签：数据结构与算法 python

原文链接：http://www.cnblogs.com/houjun/p/10454166.html

本文介绍了一种计算两个边界框（BoundingBox）交并比（IOU）的方法，使用numpy进行高效运算，确保了输入边界框坐标尺寸正确，并通过实例代码展示了计算过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Bounding Box的数据结构为（xmin,ymin,xmax,ymax）

输入：box1,box2

输出：IOU值

import numpy as np
def iou(box1,box2):
　　  assert box1.size()==4 and box2.size()==4,"bounding box coordinate size must be 4"
      bxmin = np.max(box1[0],box2[0])
      bymin = np.max(box1[1],box2[1])
      bxmax = np.min(box1[2],box2[2])
      bymax = np.min(box1[3],box2[3])
      bwidth = bxmax-bxmin
      bhight = bymax-bxmin
      inter = bwidth*bhight
      union = (box1[2]-box1[0])*(box1[3]-box1[1])+(box2[2]-box2[0])*(box2[3]-box2[1])-inter
      return inter/union