環境:OpenCV3.2.0 + VS2017
91、合并Y方向重疊的輪廓
以輪廓的最小垂直外接矩形框的y為依據,合并y重疊的輪廓。
數學邏輯:幾何合并的數學表達
- 坐標系統:假設矩形由左上角坐標
(x, y)
和寬高(width, height)
定義。- 合并公式:
- 合并后左上角:
(min(x1, x2), min(y1, y2))
- 合并后右下角:
(max(x1+w1, x2+w2), max(y1+h1, y2+h2))
- 合并后尺寸:
width = max_x - min_x
,?height = max_y - min_y
91.1:簡單直接,但對于大量輪廓可能效率較低(O(n2))
91.2:使用并查集算法,效率更高
std::vector<std::vector<cv::Point>> contour_end;//最終輪廓std::vector<cv::Rect> bound_Rect_end;if (1) { // 合并 Y方向重疊的輪廓cv::Mat visual_bR = cv::Mat::zeros(480, 640, CV_8UC3);imgOriginal.copyTo(visual_bR);cv::RNG rng(12345);std::vector<std::vector<cv::Point>> mergedContours;//mergedContours = mergeOverlappingContoursY(contour_retained);mergedContours = mergeOverlappingContoursYOptimized(contour_retained); // 優化版本if (debug) cout << "Y方向重疊合并后 mergedContours.size() = " << mergedContours.size() << endl;if (debug) {for (size_t i = 0; i < mergedContours.size(); i++) {cv::Scalar color = cv::Scalar(rng.uniform(0, 256), rng.uniform(0, 256), rng.uniform(0, 256));cv::drawContours(visual_bR, mergedContours, i, color, 2);}if (debug) cv::imshow("Merged Contours", visual_bR);}int remainNum = 0;//剩余的輪廓數std::vector<cv::Rect> boundRect;for (int i = 0; i < mergedContours.size(); i++){std::vector<cv::Point> curContours = mergedContours.at(i);if (curContours.size() < 40) continue;boundRect.push_back(cv::boundingRect(curContours));if (debug) cv::rectangle(visual_bR, boundRect[boundRect.size() - 1].tl(), boundRect[boundRect.size() - 1].br(), cv::Scalar(0, 255, 0), 1);if (debug) cv::putText(visual_bR, std::to_string(boundRect.size() - 1), boundRect[boundRect.size() - 1].tl(), cv::FONT_HERSHEY_COMPLEX, 0.45, cv::Scalar(255, 135, 160), 1);if (debug) cv::putText(visual_bR, std::to_string(boundRect.size() - 1), boundRect[boundRect.size() - 1].br(), cv::FONT_HERSHEY_COMPLEX, 0.45, cv::Scalar(255, 135, 160), 1);//if (debug) cv::drawContours(visual_bR, curContours, i, cv::Scalar(255, 135, 160), -1, CV_AA);if (boundRect.size() - 1 >= 0) {cv::Rect curBR = boundRect.at(boundRect.size() - 1);double whRatio = curBR.width*1.0 / curBR.height;//寬高比double full = curContours.size()*1.0 / (curBR.width*curBR.height);if (debug) std::cout << "--- curBR_" << boundRect.size() - 1 << curBR << whRatio << " \tfull=" << full << std::endl;//if (whRatio > 1) continue;//寬高比不滿足要求的直接 continue//if (curBR.width > imgOriginal.cols / 3) continue;//if (curBR.height > imgOriginal.rows / 3) continue;if (curBR.width < 150) continue;//if (curBR.height > 150) continue;}cv::RotatedRect curMinRect = cv::minAreaRect(curContours);float longerSide = curMinRect.size.width > curMinRect.size.height ? curMinRect.size.width : curMinRect.size.height;float shorterSide = curMinRect.size.width < curMinRect.size.height ? curMinRect.size.width : curMinRect.size.height;double lsRatio = longerSide * 1.0 / shorterSide;//長寬比if (debug) {cv::Point2f vertices[4];curMinRect.points(vertices);for (int i = 0; i < 4; i++)line(visual_bR, vertices[i], vertices[(i + 1) % 4], cv::Scalar(80, 175, 210), 2);if (debug) cv::putText(visual_bR, std::to_string(i), vertices[1], cv::FONT_HERSHEY_COMPLEX, 0.45, cv::Scalar(80, 175, 210), 1);if (debug) cv::putText(visual_bR, std::to_string(i), vertices[3], cv::FONT_HERSHEY_COMPLEX, 0.45, cv::Scalar(80, 175, 210), 1);cv::RotatedRect curMR = curMinRect;if (debug) std::cout << i << " curMR.angle=" << curMR.angle << " \t, curMR.center=" << curMR.center << "\t, curMR.size=" << curMR.size << lsRatio << std::endl;}if (lsRatio < 6.4) continue;//if (lsRatio > 2.5) continue;remainNum++;cv::Scalar color = cv::Scalar(rng.uniform(0, 256), rng.uniform(0, 256), rng.uniform(0, 256));cv::drawContours(visual_bR, mergedContours, i, color, -1, CV_AA);//cv::drawContours(visual_bR, mergedContours, i, cv::Scalar(255, 255, 255), -1, CV_AA); //用全黑色填充contour_end.push_back(curContours);bound_Rect_end.push_back(boundRect[boundRect.size() - 1]);}if (debug) cv::putText(visual_bR, std::to_string(remainNum), cv::Point(visual_bR.cols / 3, visual_bR.rows / 3), cv::FONT_HERSHEY_COMPLEX, 1.45, cv::Scalar(80, 75, 210), 1);if (debug) cv::imshow("Merged Contours", visual_bR);//visual_bR.copyTo(canvas);}
91.1 直接根據最小垂直外接矩形框的y合并?mergeOverlappingContoursY
/*
合并 Y方向上有重疊的輪廓
contours 輸入的輪廓集合
返回:合并后的輪廓集合*/
std::vector<std::vector<cv::Point>> mergeOverlappingContoursY(const std::vector<std::vector<cv::Point>>& contours)
{if (contours.empty()) return contours;// 存儲輪廓及其邊界矩形std::vector<std::pair<cv::Rect, std::vector<cv::Point>>> contourRects;for (const auto& contour : contours) {if (!contour.empty()) {contourRects.emplace_back(cv::boundingRect(contour), contour);}}// 按 Y 坐標排序std::sort(contourRects.begin(), contourRects.end(),[](const std::pair<cv::Rect, std::vector<cv::Point>>& a,const std::pair<cv::Rect, std::vector<cv::Point>>& b) {return a.first.y < b.first.y;});// 合并重疊的輪廓std::vector<std::vector<cv::Point>> mergedContours;for (size_t i = 0; i < contourRects.size(); ++i) {cv::Rect currentRect = contourRects[i].first;std::vector<cv::Point> currentContour = contourRects[i].second;// 檢查是否已經處理過if (currentRect.width == 0 && currentRect.height == 0) {continue;}// 嘗試合并與當前輪廓在 Y 方向上有重疊的輪廓for (size_t j = i + 1; j < contourRects.size(); ++j) {cv::Rect otherRect = contourRects[j].first;// 跳過已處理的輪廓if (otherRect.width == 0 && otherRect.height == 0) {continue;}// 檢查 Y 方向是否有重疊bool yOverlap = (currentRect.y <= otherRect.y + otherRect.height) &&(currentRect.y + currentRect.height >= otherRect.y);if (yOverlap) {// 合并輪廓點currentContour.insert(currentContour.end(),contourRects[j].second.begin(),contourRects[j].second.end());//opencv中以重載運算符:從 按位或 變為 矩形合并currentRect |= otherRect; // 更新當前矩形contourRects[j].first = cv::Rect(0, 0, 0, 0); // 標記已處理}}// 將合并后的輪廓添加到結果中mergedContours.push_back(currentContour);}return mergedContours;
}
91.2?使用并查集合并 Y方向上有重疊的輪廓?mergeOverlappingContoursYOptimized
- 計算每個輪廓的邊界矩形
- 使用并查集數據結構跟蹤重疊的輪廓 (對輪廓下標做并查集)
- 根據并查集結果合并輪廓 (根據下標判斷是否屬同一集合)
- 返回合并后的輪廓
/*
使用并查集合并 Y方向上有重疊的輪廓
contours 輸入的輪廓集合
返回:合并后的輪廓集合*/
std::vector<std::vector<cv::Point>> mergeOverlappingContoursYOptimized(const std::vector<std::vector<cv::Point>>& contours)
{if (contours.empty()) return contours;// 存儲輪廓及其邊界矩形std::vector<cv::Rect> rects;for (const auto& contour : contours) {if (!contour.empty()) {rects.push_back(cv::boundingRect(contour));}}// 初始化并查集std::vector<int> parent(rects.size());for (int i = 0; i < parent.size(); ++i) {parent[i] = i;}// 查找函數auto find = [&](int x) {while (parent[x] != x) {parent[x] = parent[parent[x]]; // 路徑壓縮x = parent[x];}return x;};// 合并函數auto unite = [&](int x, int y) {int rootX = find(x);int rootY = find(y);if (rootX != rootY) {parent[rootY] = rootX;}};// 檢查重疊并合并for (size_t i = 0; i < rects.size(); ++i) {for (size_t j = i + 1; j < rects.size(); ++j) {// 檢查 Y 方向是否有重疊bool yOverlap = (rects[i].y <= rects[j].y + rects[j].height) &&(rects[i].y + rects[i].height >= rects[j].y);if (yOverlap) {unite(i, j);}}}// 根據并查集結果合并輪廓std::vector<std::vector<cv::Point>> mergedContours;std::vector<bool> processed(rects.size(), false); //標記已處理的輪廓for (size_t i = 0; i < rects.size(); ++i) {if (processed[i]) continue;int root = find(i);std::vector<cv::Point> mergedContour;for (size_t j = i; j < rects.size(); ++j) {if (find(j) == root) { //同一集合的輪廓mergedContour.insert(mergedContour.end(),contours[j].begin(),contours[j].end());processed[j] = true; //標記為已處理}}mergedContours.push_back(mergedContour);}return mergedContours;
}
拓展:重疊條件修改
【重疊y超一定比例才合并
// 計算重疊比例
float overlapHeight = std::min(currentRect.y + currentRect.height, otherRect.y + otherRect.height) - std::max(currentRect.y, otherRect.y);
float minHeight = std::min(currentRect.height, otherRect.height);
float overlapRatio = overlapHeight / minHeight;// 只有當重疊比例超過閾值時才合并
if (yOverlap && overlapRatio > 0.5) {// 合并輪廓
}
【X方向重疊條件
// 添加X方向重疊條件
bool xOverlap = (rects[i].x <= rects[j].x + rects[j].width) &&(rects[i].x + rects[i].width >= rects[j].x);
bool xOverlap = (currentRect.x <= otherRect.x + otherRect.width) &&(currentRect.x + currentRect.width >= otherRect.x);// X方向重疊
【重疊面積閾值
// 添加重疊面積閾值
float overlapArea = calculateOverlapArea(rects[i], rects[j]);
float minOverlapRatio = 0.2; // 20%重疊
if (yOverlap && overlapArea / std::min(rects[i].area(), rects[j].area()) > minOverlapRatio) {unite(i, j);
}