如何合并附近的边界框OpenCV

import cv2

def merge_bounding_boxes(bounding_boxes, overlap_threshold):
    # 根据置信度对边界框进行排序
    bounding_boxes = sorted(bounding_boxes, key=lambda x: x[2], reverse=True)
    merged_boxes = []

    while len(bounding_boxes) > 0:
        current_box = bounding_boxes[0]
        merged_boxes.append(current_box)

        remaining_boxes = []
        for box in bounding_boxes[1:]:
            overlap = calculate_overlap(current_box, box)
            if overlap < overlap_threshold:
                remaining_boxes.append(box)

        bounding_boxes = remaining_boxes

    return merged_boxes

def calculate_overlap(box1, box2):
    # 计算两个边界框的重叠度
    x1, y1, w1, h1 = box1
    x2, y2, w2, h2 = box2

    area1 = w1 * h1
    area2 = w2 * h2

    intersection_x = max(x1, x2)
    intersection_y = max(y1, y2)
    intersection_w = min(x1 + w1, x2 + w2) - intersection_x
    intersection_h = min(y1 + h1, y2 + h2) - intersection_y

    intersection_area = max(0, intersection_w) * max(0, intersection_h)
    union_area = area1 + area2 - intersection_area

    overlap = intersection_area / union_area

    return overlap

# 示例用法
bounding_boxes = [(100, 100, 50, 50), (120, 120, 40, 40), (200, 200, 60, 60)]
overlap_threshold = 0.5

merged_boxes = merge_bounding_boxes(bounding_boxes, overlap_threshold)
print(merged_boxes)