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

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

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

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

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

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<std::vector<cv::Point>> contours;
	cv::findContours(imgHSVMask, contours, cv::RETR_LIST, cv::CHAIN_APPROX_NONE);
	contours.erase(std::remove_if(contours.begin(), contours.end(), 
		[](const std::vector<cv::Point>& 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;
}

原图像为：

剔除前

剔除后