2DBox和3DBox和旋转Box的iou计算

序言

面试过程中遇到的问题，如何对2Dbox、3Dbox和旋转box求iou。

2D Box计算代码

def cal_2dbox(box1, box2):"""box format: xyxy"""x1 = max(box1[0], box2[0])y1 = max(box1[1], box2[1])x2 = min(box1[2], box2[2])y2 = min(box1[3], box2[3])union = (x2 - x1) * (y2 - y1)area1 = (box1[2]-box1[0]) * (box1[3]-box1[1])area2 = (box2[2]-box2[0]) * (box2[3]-box2[1])iou = union/(area1+area2-union)return iou

3D Box计算代码

def cal_2dbox(box1, box2):"""box format: xyxy"""x1 = max(box1[0], box2[0])y1 = max(box1[1], box2[1])x2 = min(box1[2], box2[2])y2 = min(box1[3], box2[3])union = (x2 - x1) * (y2 - y1)area1 = (box1[2]-box1[0]) * (box1[3]-box1[1])area2 = (box2[2]-box2[0]) * (box2[3]-box2[1])iou = union/(area1+area2-union)return iou

旋转 Box计算代码

def cal_skewbox(box1, box2):"""box format: cxcywh angle"""area1 = box1[2]*box1[3]area2 = box2[2]*box2[3]box1_inp = ((box1[0], box1[1]), (box1[2], box1[3]), box1[4])box2_inp = ((box2[0], box2[1]), (box2[2], box2[3]), box2[4])int_pts = cv2.rotatedRectangleIntersection(tuple(box1_inp), tuple(box2_inp))[1]#order_pts = cv2.convexHull(int_pts, returnPoints=True)union = cv2.contourArea(int_pts)iou = union/(area1+area2-union)return iou

结果

if __name__ == "__main__":#2Dbox1_2d = [50, 100, 100, 200]box2_2d = [50, 50, 100, 200]iou_2d = cal_2dbox(box1_2d, box2_2d)print("二维box的计算结果:", iou_2d)#3Dbox1_3d = [50, 100, 30, 100, 200, 60]box2_3d = [50, 50, 30, 100, 200, 90]iou_3d = cal_3dbox(box1_3d, box2_3d)print("三维box的计算结果:", iou_3d)#skewbox1_skew = [50, 100, 100, 200, 0]box2_skew = [50, 100, 100, 200, 90]iou_skew = cal_skewbox(box1_skew, box2_skew)print("旋转box的计算结果:", iou_skew)

终端结果：

二维box的计算结果: 0.6666666666666666
三维box的计算结果: 0.3333333333333333
旋转box的计算结果: 0.3333333333333333

2DBox和3DBox和旋转Box的iou计算相关推荐

计算机视觉：两个旋转检测框bbox的IoU计算
计算机视觉:两个旋转检测框bbox的IoU计算 IoU简介两个旋转检测框bbox的IoU计算 IoU简介在目标检测中,经常会使用到IoU.IoU 的全称为交并比(Intersection over ...
[CV] Rotated IoU 计算旋转矩形之间的重叠面积
[CV] Rotated IoU 计算旋转矩形之间的重叠面积文章目录 [CV] Rotated IoU 计算旋转矩形之间的重叠面积简介旋转包围盒的编码方式矢量的旋转公式包围盒转化为角点代码 ...
python绘制四边形_python shapely.geometry.polygon任意两个四边形的IOU计算实例
在目标检测中一个很重要的问题就是NMS及IOU计算,而一般所说的目标检测检测的box是规则矩形框,计算IOU也非常简单,有两种方法: 1. 两个矩形的宽之和减去组合后的矩形的宽就是重叠矩形的宽,同比重 ...
yolov3前向传播（三）-- 坐标转换,iou计算,权重加载,图片显示
坐标转换,iou计算,权重加载,图片显示一.坐标转换 1.分析 2.实现 # =====================================坐标值转化函数================ ...
python图片截取斜四边形_python shapely.geometry.polygon任意两个四边形的IOU计算实例...
在目标检测中一个很重要的问题就是NMS及IOU计算,而一般所说的目标检测检测的box是规则矩形框,计算IOU也非常简单,有两种方法: 1. 两个矩形的宽之和减去组合后的矩形的宽就是重叠矩形的宽,同比重 ...
python中geometry_python shapely.geometry.polygon任意两个四边形的IOU计算实例
在目标检测中一个很重要的问题就是NMS及IOU计算,而一般所说的目标检测检测的box是规则矩形框,计算IOU也非常简单,有两种方法: 1. 两个矩形的宽之和减去组合后的矩形的宽就是重叠矩形的宽,同比重 ...
车流量检测实现：多目标追踪、卡尔曼滤波器、匈牙利算法、SORT/DeepSORT、yoloV3、虚拟线圈法、交并比IOU计算
日萌社人工智能AI:Keras PyTorch MXNet TensorFlow PaddlePaddle 深度学习实战(不定时更新) CNN:RCNN.SPPNet.Fast RCNN.Faste ...
PLC实验—西门子S7 1200读取旋转编码器数据并计算电机转速
PLC实验-西门子S7 1200读取旋转编码器数据并计算电机转速注意PTO控制步进电机实验博途软件需要V14版本,不然没有PTO功能块软件的下载请点击下方百度网盘的链接链接:https://pa ...
matlab实现iou计算,python实现IOU计算案例
计算两个矩形的交并比,通常在检测任务里面可以作为一个检测指标.你的预测bbox和groundtruth之间的差异,就可以通过IOU来体现.很简单的算法实现,我也随便写了一个,嗯,很简单. 1. 使用时 ...

2DBox和3DBox和旋转Box的iou计算

序言

2D Box计算代码

3D Box计算代码

旋转 Box计算代码

结果

2DBox和3DBox和旋转Box的iou计算相关推荐

最新文章

热门文章