OpenCV--填充图像孔洞

男娘i 2023-05-29 06:50 16阅读 0赞

1、说明

1）、要填充的区域需要连通；  
2）、图像前后景的像素值差值最好比较大；

2、代码

#include "opencv2/opencv.hpp"
    #include "vector"
    
    using namespace std;
    using namespace cv;
    
    void main(int , char **)
    { 
        //加载原图
        cv::Mat inputImg = cv::imread("4.jpg");
    
        //将原图灰度化
        cv::Mat srcGray;
        cv::cvtColor(inputImg, srcGray, CV_RGB2GRAY);
        imshow("srcGray", srcGray);
    
        int minPixValue; //孔洞边界处像素的最小值
        vector<int > v;  //存储孔洞边界处较小的像素值
    
        //遍历像素
        for (int i = 0; i < srcGray.rows; i++)
        { 
            uchar *p = srcGray.ptr<uchar>(i);
    
            for (int j = 0; j < srcGray.cols - 1; j++)
            { 
                //当像素值在10-60之间且同一行的相邻两个位置的像素值差值在10-30之间时保存像素的位置
                if (p[j] < 60 &&  p[j] > 10 &&
                        p[j + 1] < 60 &&  p[j + 1] > 10 &&
                        abs(p[j] - p[j + 1]) > 10  &&
                        abs(p[j] - p[j + 1]) < 30)
                { 
                    //p[j] < p[j + 1]说明图像由黑向白过渡，即圆的边界处；p[j] > p[j + 1]同理
                    if (p[j] < p[j + 1])
                        v.push_back(p[j]);
                    else
                        v.push_back(p[j + 1]);
                }
            }
        }
    
        //找出整幅图中孔洞边界处像素的最小值，并传递给threshold函数
        minPixValue = v[0];
        for (size_t i = 0; i < v.size(); i++)
        { 
            if (minPixValue < v[i])
                minPixValue = v[i];
        }
    
        Mat thresholdImg;
        Mat srcGray1 = srcGray.clone();
    
        // Set values below minPixValue to 255.
        threshold(srcGray1, thresholdImg, minPixValue, 255, THRESH_BINARY_INV);
        imshow("thresholdImg", thresholdImg);
    
        // Floodfill from point (0, 0)
        Mat floodfillImg = thresholdImg.clone();
        floodFill(floodfillImg, cv::Point(0, 0), Scalar(255));
        imshow("floodfillImg", floodfillImg);
    
        //与漫水填充后连通域的位置一一对应，将中间的孔洞像素值设为18
        for (int i = 0; i < floodfillImg.rows; i++)
        { 
            uchar *p1 = floodfillImg.ptr<uchar>(i);
            uchar *p2 = srcGray.ptr<uchar>(i);
    
            for (int j = 0; j < floodfillImg.cols; j++)
            { 
                if (p1[j] < 200)
                { 
                    p2[j] = 18;
                }
            }
        }
    
        //显示处理结果
        imshow("srcGray", srcGray);
    
        waitKey(0);
    }

3、结果

原图：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjcwNDA5MA_size_16_color_FFFFFF_t_70]  
填充后的效果图：（注意：填充后的效果为原图的灰度图，如何转化为原彩色图还没研究，有做出来的小伙伴可以分享一下）  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjcwNDA5MA_size_16_color_FFFFFF_t_70 1]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjcwNDA5MA_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20191105131820882.jpg?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjcwNDA5MA==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjcwNDA5MA_size_16_color_FFFFFF_t_70 1]: https://img-blog.csdnimg.cn/2019110513185474.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80MjcwNDA5MA==,size_16,color_FFFFFF,t_70