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
面积判断
最新推荐文章于 2019-09-27 10:42:42 发布