Python中利用iou()函数计算物体边界框的重叠度

在 Python 中，可以使用 iou() 函数来计算物体边界框的重叠度。IOU (Intersection over Union) 是一种常用的度量方法，用于衡量两个边界框之间的重叠程度。

首先，我们需要导入所需的库和模块：

import numpy as np

然后，我们可以定义一个 iou() 函数，该函数接受两个边界框的坐标作为参数，并返回它们的重叠度。

def iou(box1, box2):
    # 计算边界框的面积
    area_box1 = (box1[2] - box1[0] + 1) * (box1[3] - box1[1] + 1)
    area_box2 = (box2[2] - box2[0] + 1) * (box2[3] - box2[1] + 1)

    # 计算相交区域的坐标
    x1 = max(box1[0], box2[0])
    y1 = max(box1[1], box2[1])
    x2 = min(box1[2], box2[2])
    y2 = min(box1[3], box2[3])

    # 计算相交区域的面积
    inter_area = max(0, x2 - x1 + 1) * max(0, y2 - y1 + 1)

    # 计算并返回重叠度
    iou = inter_area / float(area_box1 + area_box2 - inter_area)
    return iou

接下来，让我们使用一些示例来演示如何使用这个 iou() 函数。

# 定义两个边界框的坐标 [x1, y1, x2, y2]
box1 = [0, 0, 10, 10]
box2 = [5, 5, 15, 15]

# 计算边界框的重叠度
iou_score = iou(box1, box2)
print("IOU score:", iou_score)

在这个例子中，我们有两个边界框 box1 和 box2，它们的左上角和右下角坐标分别为 [0, 0, 10, 10] 和 [5, 5, 15, 15]。使用 iou() 函数，我们计算出它们的重叠度得分并打印出来。

输出结果将为：

IOU score: 0.25

这表示两个边界框之间的重叠度为 25%。

将这个例子作为示例，你可以使用 iou() 函数计算任意两个物体边界框的重叠度。请确保传递的边界框参数是以 [x1, y1, x2, y2] 的形式组织的，并将结果进行适当的解释和使用。