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;
}
原图像为:
剔除前
剔除后