c/c++语言开发共享C++ Opencv imfill孔洞填充函数的实现思路与代码

目录

  mat imfill(mat cop)  {  	mat fcop;  	cop.copyto(fcop);  	vector<point> upp;//定义四连通点集，有必要可以是八连通  	upp.push_back(point(-1, 0));  	upp.push_back(point(0, -1));  	upp.push_back(point(0, 1));  	upp.push_back(point(1, 0));  	//upp.push_back(point(1, 1));  	//upp.push_back(point(-1,-1));  	//upp.push_back(point(-1, 1));  	//upp.push_back(point(1, -1));    	vector<vector<vector<point>>> lenm;//三维point向量 lenm.size()是连通域的个数  	/*  	int impixel_sum = 0;  	for (int j = 0,j<lenm[i].size();j++)  	{  	    impixel_sum+= lenm[i][j].size();  	}  	//这段循环 表示第i个连通域中 像素点的个数。  	*/  	vector<vector<point>> numim;  	vector<point> ssinum;  	vector<vector<int>> flag;  	vector<int> ce;    	int nmss = 0;//连通域的个数；  	int nums = 0;//中间变量 用来存储 lenm.size();即 在程序运行过程中 nums始终等于 lenm[i][j][k] 中的j 的 值的大小;  	int s1 = 0;  	//标志位 ，每次区域生长后 符合条件的像素个数，当第i个连通域，在经过第j次生长后，s1=lenm[i][j].size(),  	//若s1==0,表示生长结束,不再有符合条件的点，第i连通域中的所有点都已经找到。    	for (int row = 0; row < fcop.rows; row++)  	{  		for (int col = 0; col < fcop.cols; col++)  		{  			if (fcop.at<uchar>(row, col) == 0)  			{  				ce.push_back(1);  				flag.push_back(ce);  				//vector<vector<point>> numim;  				//vector<point> ssinum;  				ssinum.push_back(point(col, row));  				numim.push_back(ssinum);  				fcop.at<uchar>(row, col) = 255;  				ssinum.clear();  				s1 = 1;  				while (s1 > 0)  				{  					//ce.push_back(1);  					//flag.push_back(ce);  					//vector<point> ssinum;  					for (int i = 0; i < numim[nums].size(); i++)  					{  						for (int j = 0; j < upp.size(); j++)  						{  							int x = numim[nums][i].x + upp[j].x;  							int y = numim[nums][i].y + upp[j].y;  							if (x >= 0 && y >= 0 && x < fcop.cols && y < fcop.rows)  							{  								if (fcop.at<uchar>(y, x) == 0)  								{  									ssinum.push_back(point(x, y));  									fcop.at<uchar>(y, x) = 255;  								}  							}    							if (x == 0 || y == 0 || x == fcop.cols - 1 || y == fcop.rows - 1)  							{  								flag[nmss][0] = 0;    							}  						}  					}  					//flag.push_back(ce);  					numim.push_back(ssinum);  					s1 = ssinum.size();  					nums++;  					ssinum.clear();  					/*ce.clear();*/  				}  				nums = 0;  				lenm.push_back(numim);  				numim.clear();  				nmss++;  				ce.clear();  			}    		}  	}  	//imshow("1",fcop);  	mat ffcop;  	cop.copyto(ffcop);  	//ffcop = mat::zeros(cop.size(),cop.type());  	for (int i = 0; i < flag.size(); i++)  	{  		if (flag[i][0] == 1)  		{  			for (int j = 0; j < lenm[i].size(); j++)  			{  				for (int k = 0; k < lenm[i][j].size(); k++)  				{  					int x = lenm[i][j][k].x;  					int y = lenm[i][j][k].y;  					ffcop.at<char>(y, x) = 255;  				}  			}  		}  	}  	return ffcop;  }  

  #include<opencv2/opencv.hpp>  #include<iostream>    #include"imfill.h"    using namespace std;  using namespace cv;    mat src;  vector<vector<point>>  lunk;  vector<vec4i> level;    //rng rn;  int main()  {  	src = imread("5.jpg");  	//imshow("万丈高楼第一步",src);    	mat dst, gray, erzhi;  	blur(src, dst, size(3, 3), point(-1, -1));  	//imshow("均值滤波",dst);    	cvtcolor(dst, gray, color_bgr2gray);  	//imshow("灰度图",gray);    	canny(gray, erzhi, 100, 200, 3, false);  	//imshow("边缘检测",erzhi);    	mat holef;  	holef = imfill(erzhi);  	imshow("填洞", holef);  	  	waitkey(0);  	return 0;  }