如何基于已有标记在OpenCV中提取同标签内的连通组件？

好问题！确实，cv::connectedComponents() 默认会把图像中所有非零像素当作同一类来查找连通域，没法直接利用图像已有的标记信息进行分组处理。不过我们可以通过分标记掩码处理的方式来实现需求，下面是具体步骤和示例代码：

实现思路 #

• 第一步：提取图像中所有非背景的唯一标记值（比如示例里的1和2）
• 第二步：针对每个标记值，生成只包含该标记像素的掩码图像
• 第三步：对每个掩码单独调用 cv::connectedComponents()，再将结果映射回原标记区域，区分同一标记下的不同连通组件

完整示例代码 #

#include <opencv2/opencv.hpp>
#include <unordered_set>
#include <iostream>

int main() {
    // 初始化用户提供的测试图像
    cv::Mat test = cv::Mat::zeros(1, 4, CV_32S);
    test.at<int>(0, 0) = 1;
    test.at<int>(0, 1) = 2;
    test.at<int>(0, 2) = 0;
    test.at<int>(0, 3) = 1;
    cv::Mat test_8u;
    test.convertTo(test_8u, CV_8U);

    // 存储最终的分组连通组件结果
    cv::Mat result = cv::Mat::zeros(test.size(), CV_32S);
    int current_label_offset = 1; // 避免和原标记混淆，偏移量从1开始

    // 第一步：提取所有唯一的非背景标记
    std::unordered_set<int> unique_labels;
    for (int y = 0; y < test.rows; ++y) {
        for (int x = 0; x < test.cols; ++x) {
            int label = test.at<int>(y, x);
            if (label != 0) {
                unique_labels.insert(label);
            }
        }
    }

    // 第二步：遍历每个标记，生成掩码并查找连通组件
    for (int label : unique_labels) {
        cv::Mat mask = (test == label); // 生成当前标记的掩码

        // 第三步：对掩码调用connectedComponents
        cv::Mat comps;
        int num_comps = cv::connectedComponents(mask, comps, 8, CV_32S);

        // 将当前标记的连通组件结果映射到最终结果图
        for (int y = 0; y < comps.rows; ++y) {
            for (int x = 0; x < comps.cols; ++x) {
                int comp_label = comps.at<int>(y, x);
                if (comp_label != 0) { // 跳过背景
                    result.at<int>(y, x) = current_label_offset + comp_label - 1;
                }
            }
        }

        // 更新偏移量，确保不同标记的连通组件标签不重复
        current_label_offset += num_comps - 1;
    }

    // 输出结果验证
    std::cout << "原标记图像：" << std::endl;
    std::cout << test << std::endl;
    std::cout << "\n按标记分组的连通组件结果：" << std::endl;
    std::cout << result << std::endl;

    return 0;
}

代码说明 #

• 我们先用std::unordered_set提取图像中所有非零的唯一标记，避免重复处理
• 对每个标记生成的掩码，cv::connectedComponents()会在该掩码内查找连通域，这样就实现了“仅在相同标记像素范围内查找”的需求
• 最终结果里，同一原标记下的不同连通组件会被赋予不同的新标签（比如示例里原标记1的两个孤立像素会被标记为1和2，原标记2的像素被标记为3）

运行这段代码后，你会得到如下输出：

原标记图像：
[1, 2, 0, 1]

按标记分组的连通组件结果：
[1, 3, 0, 2]

内容的提问来源于stack exchange，提问作者Viktor Petukhov