如何在 OpenCV 中組合邊界框（Python）

Question

您好，我有一個rect數組，代表檢測到的對象的邊界框。 通常，這些矩形重疊，我想將所有相交的矩形合並為一個大矩形（例如，如果有兩個單獨的簇，則屏幕上有多個矩形）。 有沒有辦法做到這一點？

我已經嘗試過cv2.groupRectangles(rect, 1, .5) ，但是矩形簇在中間被分組為一個與原始矩形大小相同的矩形。

編輯：

這是沒有分組的圖片：

這是帶有分組的圖片和下面描述的方法

為了更清楚地說明我在尋找什么，我只想在各個塊上放置紅色框（或者最好是在整行上放置一個框），而不是在中間，這樣可以壓縮我需要的框數量進行計算以加速程序

Answer 1

您將通過分組矩形到 go 並使用源矩形檢查它們，如果它們相交而任何更改（請參閱flag ），則重寫它們的邊界：

def get_intersection_area(box_a, box_b):
    x_a = max(box_a[0], box_b[0])
    y_a = max(box_a[1], box_b[1])
    x_b = min(box_a[2], box_b[2])
    y_b = min(box_a[3], box_b[3])
    return max(0, x_b - x_a) * max(0, y_b - y_a)


groupped = cv2.groupRectangles(rect, 1, 0.5)[0]
groupped = [(x, y, x+w, y+h) for x, y, w, h in groupped]
rect = [(x, y, x+w, y+h) for x, y, w, h in rect]
flag = True
while flag:
    flag = False
    for i in range(len(groupped)):
        for r in rect:
            if 0 < get_intersection_area(groupped[i], r) <\
                    (r[2]-r[0])*(r[3]-r[1]):
                groupped[i] = (
                    min(groupped[i][0], r[0]), min(groupped[i][1], r[1]),
                    max(groupped[i][2], r[2]), max(groupped[i][3], r[3]),)
                flag = True

（我已經將盒子從 tlwh 格式轉換為 tlrb，因為它對我來說更方便）

這是它現在給出的：

要保留不相交的矩形，請不要使用cv2.groupRectangles()
只說groupped = np.copy(rect) ，盡管它可能會以這種方式做額外的工作並且需要一些優化。

Answer 2

一種簡單的方法是繪制所有填充在遮罩上的矩形。 然后對該蒙版執行 findContours 以找到組合的形狀。 要使結果成為正方形，您可以使用 boundingRect 來查找 x、y、寬度和高度。 請注意，這完全結合了所有矩形，而不僅僅是重疊部分。

您可以在此處找到用於組合矩形示例的示例代碼