目标检测计算两个标注框的交并比(python代码)
Posted 悠悠青青
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了目标检测计算两个标注框的交并比(python代码)相关的知识,希望对你有一定的参考价值。
def iou(box1, box2):
'''
两个框(二维)的 iou 计算
box:[xmin,ymin,xmax,ymax], 依次为左上、右下坐标
@return: IOU值
'''
h = max(0, min(box1[2], box2[2]) - max(box1[0], box2[0]))
w = max(0, min(box1[3], box2[3]) - max(box1[1], box2[1]))
area_box1 = ((box1[2] - box1[0]) * (box1[3] - box1[1]))
area_box2 = ((box2[2] - box2[0]) * (box2[3] - box2[1]))
inter = w * h
union = area_box1 + area_box2 - inter
iou = inter / union
return iou
以上是关于目标检测计算两个标注框的交并比(python代码)的主要内容,如果未能解决你的问题,请参考以下文章