面积判断

void RemoveSmallRegion(Mat &src, Mat &dst,
					   int AreaLimit,   //= 200,
					   int checkMode ,//= 1,  // 0: delte black area; 1: delte white area
					   int NeihborMode  // = 0 // 0: 4_continue  1: 8_continue;
					   )
{
	int RemoveCount = 0;
    
	Mat PointLabel = Mat::zeros(src.size(), CV_8UC1); //construct image

	if(checkMode == 1) 
	{
		cout<<"Mode: delete small area";
		for(int i = 0; i < src.rows; ++i)
		{
			uchar *iData = src.ptr<uchar>(i);
			uchar *iLabel = PointLabel.ptr<uchar>(i);
			for(int j = 0; j <src.cols; ++j)
			{
				if( iData[j] < 10)   //each for every pixel 
					iLabel[j] = 3;
			}
		}
	}else
	{
		cout<<"Mode: delete hole";
	    for(int i = 0; i < src.rows; ++i)
		{
			uchar *iData = src.ptr<uchar>(i);
			uchar *iLabel = PointLabel.ptr<uchar>(i);
			for(int j = 0; j <src.cols; ++j)
			{
				if( iData[j] > 10)
					iLabel[j] = 3;
			}
		}
	}

	vector<Point2i> NeihborPos;
	NeihborPos.push_back(Point2i(-1, 0));
	NeihborPos.push_back(Point2i(1, 0));
	NeihborPos.push_back(Point2i(0, -1));
	NeihborPos.push_back(Point2i(0, 1));
	if(NeihborMode == 1)
	{
		cout<<"Neighbor mode: 8 connect ."<<endl;  
        NeihborPos.push_back(Point2i(-1, -1));  
        NeihborPos.push_back(Point2i(-1, 1));  
        NeihborPos.push_back(Point2i(1, -1));  
        NeihborPos.push_back(Point2i(1, 1));  

	}
	else
		cout<<"Neighbor mode: 4 connect."<<endl; 

    int NeihborCount=4+4*NeihborMode;  
    int CurrX=0, CurrY=0;  
    //开始检测  
    for(int i = 0; i < src.rows; ++i)    
    {    
        uchar* iLabel = PointLabel.ptr<uchar>(i);  
        for(int j = 0; j < src.cols; ++j)    
        {    
            if (iLabel[j] == 0)    
            {    
                //********开始该点处的检查**********  
                vector<Point2i> GrowBuffer;      //堆栈,用于存储生长点  
                GrowBuffer.push_back( Point2i(j, i) );  
                PointLabel.at<uchar>(i, j)=1;  
                int CheckResult=0;             //用于判断结果(是否超出大小),0为未超出,1为超出  
  
                for ( int z=0; z<GrowBuffer.size(); z++ )  
                {  
                    for (int q=0; q<NeihborCount; q++)     //检查四个邻域点  
                    {  
                        CurrX=GrowBuffer.at(z).x+NeihborPos.at(q).x;  
                        CurrY=GrowBuffer.at(z).y+NeihborPos.at(q).y;  
                        if (CurrX>=0 && CurrX<src.cols && CurrY>=0 && CurrY<src.rows)  //防止越界  
                        {  
                            if ( PointLabel.at<uchar>(CurrY, CurrX)==0 )  
                            {  
                                GrowBuffer.push_back( Point2i(CurrX, CurrY) );  //邻域点加入buffer  
                                PointLabel.at<uchar>(CurrY, CurrX)=1;           //更新邻域点的检查标签,避免重复检查  
                            }  
                        }  
                    }  
  
                }  
                if (GrowBuffer.size()>AreaLimit) CheckResult=2;                 //判断结果(是否超出限定的大小),1为未超出,2为超出  
                else {CheckResult=1;   RemoveCount++;}  
                for (int z=0; z<GrowBuffer.size(); z++)                         //更新Label记录  
                {  
                    CurrX=GrowBuffer.at(z).x;   
                    CurrY=GrowBuffer.at(z).y;  
                    PointLabel.at<uchar>(CurrY, CurrX) += CheckResult;  
                }  
                //********结束该点处的检查**********  
            }    
        }    
    }    
  
    checkMode=255*(1-checkMode);  
    //开始反转面积过小的区域  
    for(int i = 0; i < src.rows; ++i)    
    {    
        uchar* iData = src.ptr<uchar>(i);  
        uchar* iDstData = dst.ptr<uchar>(i);  
        uchar* iLabel = PointLabel.ptr<uchar>(i);  
        for(int j = 0; j < src.cols; ++j)    
        {    
            if (iLabel[j] == 2)    
            {    
                iDstData[j] = checkMode;   
            }    
            else if(iLabel[j] == 3)  
            {  
                iDstData[j] = iData[j];  
            }  
        }    
    }   
      
    cout<<RemoveCount<<" objects removed."<<endl;  
}


int _tmain(int argc, _TCHAR* argv[])
{
	Mat image = imread("C:\\Users\\z12073245\\Desktop\\13.jpg");
	namedWindow("[原始图]");
	namedWindow("[效果图]");
	imshow("[原始图]", image);
	
	//Mat element = getStructuringElement(MORPH_RECT, Size(3,3));
    //erode(image, image, element, Point(-1, -1), 1);

	//morphologyEx(image,image, MORPH_ERODE, element, Point(-1, -1), 1);
   
	cvtColor(image, image, COLOR_RGB2GRAY); 
	Mat dst = Mat::zeros(image.size(), CV_8UC1);

	RemoveSmallRegion(image, dst, 250, 1, 0);

	imshow("[效果图]", dst);
	waitKey(0);
	return 0;
}
转载自:http://lib.csdn.net/article/opencv/28353
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值