OpenCV: Слияние перекрывающихся прямоугольников

Question

При использовании OpenCV для задач обнаружения я постоянно сталкиваюсь с проблемой слияния перекрывающих ограничивающих прямоугольников; то есть, по существу, обнаруживая ограничительную рамку вокруг объединения двух перекрывающихся ограничивающих прямоугольников. Это очень часто встречается при обнаружении объектов, когда по какой-либо причине объект интереса разломается на несколько ограничивающих прямоугольников, а не на одно всеобъемлющее.

Есть несколько ответов на StackOverflow для алгоритмических решений и полезных внешних библиотек (например this, this, this), и есть также функция groupRectangles обеспечивается OpenCV (и литания связанных вопросов/ошибок: this, this, и т.д).

Я, кажется, обнаружил, что вышеупомянутые решения немного сложны для задачи, которую я пытаюсь выполнить. Многие алгоритмические решения решают эту проблему с математической точки зрения (скорее, как мысленный эксперимент), в то время как операции, подобные rect1 | rect2, могут стать очень медленными, когда число прямоугольников велико (это O (N^2) для обработки всего) и groupRectangles имеет некоторые причуды, которые делают его лишь частично эффективным. Поэтому я придумал несколько хакерское решение, которое действительно работает довольно эффективно. Я думал, что буду делиться этим ниже для всех, кому нужно быстрое решение этой общей проблемы.

Не стесняйтесь, чтобы комментировать и критиковать его.

Answer 1

void mergeOverlappingBoxes(std::vector<cv::Rect> &inputBoxes, cv::Mat &image, std::vector<cv::Rect> &outputBoxes) 
{ 
    cv::Mat mask = cv::Mat::zeros(image.size(), CV_8UC1); // Mask of original image 
    cv::Size scaleFactor(10,10); // To expand rectangles, i.e. increase sensitivity to nearby rectangles. Doesn't have to be (10,10)--can be anything 
    for (int i = 0; i < inputBoxes.size(); i++) 
    { 
     cv::Rect box = inputBoxes.at(i) + scaleFactor; 
     cv::rectangle(mask, box, cv::Scalar(255), CV_FILLED); // Draw filled bounding boxes on mask 
    } 

    std::vector<std::vector<cv::Point>> contours; 
    // Find contours in mask 
    // If bounding boxes overlap, they will be joined by this function call 
    cv::findContours(mask, contours, CV_RETR_EXTERNAL, CV_CHAIN_APPROX_SIMPLE); 
    for (int j = 0; j < contours.size(); j++) 
    { 
     outputBoxes.push_back(cv::boundingRect(contours.at(j))); 
    } 
}