opencv 二值图像剔除小连通区域

二值图像剔除小面积连通区域在二值图像连通区域分析时很有用，之前做的使用采用了for循环的形式，后来学习了c++标准库，发现可以使用vector.erase(std::remove_if())的方法， 结合lambda表达式直接剔除。

统计二值图像的连通区域通过cv::findcontours()实现，二值图像轮廓的容器是std::vector。连通区域的面积可以由函数cv::contourArea()得到。

剔除小面积连通区域后，可以使用函数cv::drawContours()函数将轮廓画出，将其第三个参数设置为-1为画所有轮廓，将其第5个参数设置为cv::FILLED设置为填充。则可以得到剔除后小面积连通后的二值图像。

#include 
#include 
#include 

int main()
{
    cv::Mat img = cv::imread("data/000000000597.jpg");
    cv::Mat imgHSV(img.size(), img.type());
    cv::cvtColor(img, imgHSV, cv::COLOR_BGR2HSV);
    cv::Mat imgHSVMask(img.size(), CV_8UC1);
    // yellow color region
    cv::inRange(imgHSV, cv::Scalar(11, 43, 46), cv::Scalar(26, 255, 255), imgHSVMask);
    
    char *win1 = "original mask";   char *win2 = "after remove";
    cv::namedWindow(win1);  cv::moveWindow(win1, 10, 10);
    cv::namedWindow(win2);  cv::moveWindow(win2, 800, 10);
    
    cv::imshow(win1, imgHSVMask);   cv::waitKey(0);     cv::destroyWindow(win1);    cv::imwrite("data/original.png", imgHSVMask);

    // 提取连通区域，并剔除小面积联通区域
    std::vector> contours;
    cv::findContours(imgHSVMask, contours, cv::RETR_LIST, cv::CHAIN_APPROX_NONE);
    contours.erase(std::remove_if(contours.begin(), contours.end(), 
        [](const std::vector& c){return cv::contourArea(c) < 30; }), contours.end());

    // 显示图像并保存
    imgHSVMask.setTo(0);
    cv::drawContours(imgHSVMask, contours, -1, cv::Scalar(255), cv::FILLED);
    cv::imshow(win2, imgHSVMask);   cv::waitKey(0); cv::destroyWindow(win2);    cv::imwrite("data/remove.png", imgHSVMask);
    return 0;
}

原图像为：

剔除前

剔除后

转载于:https://www.cnblogs.com/zi-wang/p/9800141.html