cv::connectedComponents保留面积最大的两个-Go服务器开发

要使用 cv::connectedComponents 函数并保留面积最大的两个连通域，你可以按以下步骤实现：

加载和二值化图像。
使用 cv::connectedComponents 来检测连通域。
计算每个连通域的面积，并找到最大的两个。
创建输出图像，只绘制这两个最大连通域。

以下是示例代码：

#include <opencv2/opencv.hpp>
#include <vector>
#include <algorithm>

int main() {
    // 读取图像
    cv::Mat image = cv::imread("path_to_your_image.jpg", cv::IMREAD_GRAYSCALE);
    if (image.empty()) {
        std::cerr << "Could not read the image." << std::endl;
        return -1;
    }

    // 二值化处理
    cv::Mat binaryImage;
    cv::threshold(image, binaryImage, 128, 255, cv::THRESH_BINARY);

    // 使用 connectedComponents 检测连通域
    cv::Mat labels;
    int numLabels = cv::connectedComponents(binaryImage, labels);

    // 创建一个向量来存储每个标签对应的面积
    std::vector<int> areas(numLabels, 0);

    // 计算每个连通区域的面积
    for (int i = 0; i < labels.rows; ++i) {
        for (int j = 0; j < labels.cols; ++j) {
            int label = labels.at<int>(i, j);
            if (label > 0) { // 忽略背景（标签为0）
                areas[label]++;
            }
        }
    }

    // 找到最大的两个连通区域的索引
    std::vector<std::pair<int, int>> areaIndexPairs;
    
    for (int i = 1; i < areas.size(); ++i) {
        areaIndexPairs.emplace_back(areas[i], i);
    }

   // 按照面积降序排序，并获取前两个最大值
   std::sort(areaIndexPairs.begin(), areaIndexPairs.end(), std::greater<>());

   int largestLabel = areaIndexPairs[0].second;
   int secondLargestLabel = areaIndexPairs.size() > 1 ? areaIndexPairs[1].second : -1;

   // 创建输出图像，只绘制最大两个连通域
   cv::Mat outputImage = cv::Mat::zeros(binaryImage.size(), CV_8UC1);

   if (largestLabel != -1) {
       outputImage.setTo(cv::Scalar(255), labels == largestLabel); 
   }

   if (secondLargestLabel != -1) {
       outputImage.setTo(cv::Scalar(128), labels == secondLargestLabel); 
   }

   // 显示结果
   cv::imshow("Original Image", image);
   cv::imshow("Binary Image", binaryImage);
   cv::imshow("Largest Two Connected Components", outputImage);

   cv::waitKey(0);
   
   return 0;
}