一、对于二值图,0代表黑色,255代表白色。去除小连通区域与孔洞,小连通区域用8邻域,孔洞用4邻域。
函数名字为:void RemoveSmallRegion(Mat &Src, Mat &Dst,int AreaLimit, int CheckMode, int NeihborMode)
CheckMode: 0代表去除黑区域,1代表去除白区域; NeihborMode:0代表4邻域,1代表8邻域;
如果去除小连通区域CheckMode=1,NeihborMode=1去除孔洞CheckMode=0,NeihborMode=0
记录每个像素点检验状态的标签,0代表未检查,1代表正在检查,2代表检查不合格(需要反转颜色),3代表检查合格或不需检查 。
1.先对整个图像扫描,如果是去除小连通区域,则将黑色的背景图作为合格,像素值标记为3,如果是去除孔洞,则将白色的色素点作为合格,像素值标记为3。
2.扫面整个图像,对图像进行处理。
- void RemoveSmallRegion(Mat &Src, Mat &Dst,int AreaLimit, int CheckMode, int NeihborMode)
- {
- int RemoveCount = 0;
-
-
- Mat PointLabel = Mat::zeros(Src.size(), CV_8UC1);
- if (CheckMode == 1)
- {
- cout << "去除小连通域.";
- for (int i = 0; i < Src.rows; i++)
- {
- for (int j = 0; j < Src.cols; j++)
- {
- if (Src.at<uchar>(i, j) < 10)
- {
- PointLabel.at<uchar>(i, j) = 3;
- }
- }
- }
- }
- else
- {
- cout << "去除孔洞";
- for (int i = 0; i < Src.rows; i++)
- {
- for (int j = 0; j < Src.cols; j++)
- {
- if (Src.at<uchar>(i, j) > 10)
- {
- PointLabel.at<uchar>(i, 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邻域." << 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邻域." << endl;
- int NeihborCount = 4 + 4 * NeihborMode;
- int CurrX = 0, CurrY = 0;
-
- for (int i = 0; i < Src.rows; i++)
- {
- for (int j = 0; j < Src.cols; j++)
- {
- if (PointLabel.at<uchar>(i, j) == 0)
- {
- vector<Point2i>GrowBuffer;
- GrowBuffer.push_back(Point2i(j, i));
- PointLabel.at<uchar>(i, j) = 1;
- int CheckResult = 0;
-
-
- 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));
- PointLabel.at<uchar>(CurrY, CurrX) = 1;
- }
- }
- }
- }
- if (GrowBuffer.size()>AreaLimit)
- CheckResult = 2;
- else
- {
- CheckResult = 1;
- RemoveCount++;
- }
-
-
- for (int z = 0; z < GrowBuffer.size(); z++)
- {
- 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)
- {
- for (int j = 0; j < Src.cols; ++j)
- {
- if (PointLabel.at<uchar>(i,j)==2)
- {
- Dst.at<uchar>(i, j) = CheckMode;
- }
- else if (PointLabel.at<uchar>(i, j) == 3)
- {
- Dst.at<uchar>(i, j) = Src.at<uchar>(i, j);
-
- }
- }
- }
- cout << RemoveCount << " objects removed." << endl;
- }
调用函数:dst是原来的二值图。
Mat erzhi1 = Mat::zeros(srcImage.rows, srcImage.cols, CV_8UC1);
RemoveSmallRegion(dst, erzhi,100, 1, 1);
RemoveSmallRegion(erzhi, erzhi,100, 0, 0);
imshow("erzhi1", erzhi);
和之前的图像相比
原文链接:http://blog.csdn.net/dajiyi1998/article/details/60601410#