c++ - 在opencv和c ++中合并重叠边界框的函数问题

Question

我在为对象检测任务的合并边界框编写适当的函数时遇到问题。我将在下面报告我的尝试，它在detectMultiScale调用的输出中输入Rect对象的向量，重叠阈值和容差系数以合并足够接近但没有真正重叠的框，而不是删除“重叠者”将他们的工会视为更可靠的代表。具体来说，它应该扫描向量，并且对于每个提取的 Rect r，扫描剩余的向量：如果 r 中包含另一个 Rect s（相对于阈值，“超过”条件），则从正方形中删除 s。此外，如果 s 不包含但足够接近（相对于 tol，“接近”条件），则 r 在向量中被替换为 r 和 s 之间的并集，和 s 被删除。如果没有任何条件发生，它只会继续循环。问题是此功能在擦除框时过于激进，使设置如所附输出图像比较所示（使用 th = 0.8 和 tol = 0.8 获得），我无法理解我做错了什么。

vector<Rect> mergeBoxes(vector<Rect> squaresc, double th, double tol) {
    vector<Rect> squares = vector<Rect>(squaresc);
    int q = 0;
    while (q < squares.size()) {
        Rect r = Rect(squares[q]);
        int w = 0;
        while (w < squares.size()) {
            if (w == q) {
                break;
            }
            Rect s = Rect(squares[w]);
            Rect is = r & s;
            int inters_area = is.area();
            bool check = ((inters_area / r.area() < th) && (inters_area / s.area() > th));
            bool close = (r.area() + s.area() >= tol*(s|r).area());
            if (check) {
                squares.erase(squares.begin() + w);
            }
            else if (close) {
                squares[q] = r | s;
                squares.erase(squares.begin() + w);

            }
            else {
                w++;
            }
        }
        q++;
    }
    return squares;
}

不使用 mergeBoxes 和使用 mergeBoxes 后的输出

标签： c++opencvobject-detectionbounding-boxcascade-classifier