基于灰度图或者RGB的区域生长算法(C++)

在这里插入图片描述

1.算法原理

区域生长是根据预先定义的生长准则将像素或子区域组合为更大区域的过程。基本方法是从一组“种子”点开始(原点),将与种子相似的临近像素(在特定范围内的灰度或颜色)添加到种子栈中,不断迭代,生成一大片区域。严谨的数学定义可以查看冈萨雷斯的数字图像处理。

2.算法实现

算法的步骤如下:

  1. 创建一个与原图像大小相同的空白图像
  2. 将种子点存入vector中,vector中存储待生长的种子点
  3. 依次弹出种子点并判断种子点如周围8领域的关系(生长规则)并与最大与最小阈值进行比较,符合条件则作为下次生长的种子点
  4. vector中不存在种子点后就停止生长

3. 灰度图

/**
 * @brief 区域生长算法,输入图像应为灰度图像
 * @param srcImage 区域生长的源图像
 * @param pt 区域生长点
 * @param ch1Thres 通道的生长限制阈值,临近像素符合±chxThres范围内才能进行生长
 * @param ch1LowerBind 通道的最小值阈值
 * @param ch1UpperBind 通道的最大值阈值,在这个范围外即使临近像素符合±chxThres也不能生长
 * @return 生成的区域图像(二值类型)
 */
Mat RegionGrow(Mat srcImage, Point pt, int ch1Thres,int ch1LowerBind=0,int ch1UpperBind=255)
{
    
    
    Point pToGrowing;                       //待生长点位置
    int pGrowValue = 0;                             //待生长点灰度值
    Scalar pSrcValue = 0;                               //生长起点灰度值
    Scalar pCurValue = 0;                               //当前生长点灰度值
    Mat growImage = Mat::zeros(srcImage.size(), CV_8UC1);   //创建一个空白区域,填充为黑色
    //生长方向顺序数据
    int DIR[8][2] = {
    
    {
    
    -1,-1}, {
    
    0,-1}, {
    
    1,-1}, {
    
    1,0}, {
    
    1,1}, {
    
    0,1}, {
    
    -1,1}, {
    
    -1,0}};
    vector<Point> growPtVector;                     //生长点栈
    growPtVector.push_back(pt);                         //将生长点压入栈中
    growImage.at<uchar>(pt.y, pt.x) = 255;              //标记生长点
    pSrcValue = srcImage.at<uchar>(pt.y, pt.x);         //记录生长点的灰度值

    while (!growPtVector.empty())                       //生长栈不为空则生长
    {
    
    
        pt = growPtVector.back();                       //取出一个生长点
        growPtVector.pop_back();

        //分别对八个方向上的点进行生长
        for (int i = 0; i<9; ++i)
        {
    
    
            pToGrowing.x = pt.x + DIR[i][0];
            pToGrowing.y = pt.y + DIR[i][1];
            //检查是否是边缘点
            if (pToGrowing.x < 0 || pToGrowing.y < 0 ||
                    pToGrowing.x > (srcImage.cols-1) || (pToGrowing.y > srcImage.rows -1))
                continue;

            pGrowValue = growImage.at<uchar>(pToGrowing.y, pToGrowing.x);       //当前待生长点的灰度值
            pSrcValue = srcImage.at<uchar>(pt.y, pt.x);
            if (pGrowValue == 0)                    //如果标记点还没有被生长
            {
    
    
                pCurValue = srcImage.at<uchar>(pToGrowing.y, pToGrowing.x);
                if(pCurValue[0] <= ch1UpperBind && pCurValue[0] >= ch1LowerBind )
                {
    
    
                    if (abs(pSrcValue[0] - pCurValue[0]) < ch1Thres )                   //在阈值范围内则生长
                    {
    
    
                        growImage.at<uchar>(pToGrowing.y, pToGrowing.x) = 255;      //标记为白色
                        growPtVector.push_back(pToGrowing);                 //将下一个生长点压入栈中
                    }
                }
            }
        }
    }
    return growImage.clone();
}

4. RGB

/**
 * @brief 区域生长算法,输入图像应为三通道图像(RGB、HSV、YUV等)
 * @param srcImage 区域生长的源图像
 * @param pt 区域生长点
 * @param ch1Thres ch2Thres ch3Thres 三个通道的生长限制阈值,临近像素符合±chxThres范围内才能进行生长
 * @param ch1LowerBind ch1LowerBind ch1LowerBind 三个通道的最小值阈值
 * @param ch1UpperBind ch2UpperBind ch3UpperBind 三个通道的最大值阈值,在这个范围外即使临近像素符合±chxThres也不能生长
 * @return 生成的区域图像(二值类型)
 */
Mat RegionGrow(Mat srcImage, Point pt, int ch1Thres,int ch2Thres, int ch3Thres,
               int ch1LowerBind=0,int ch1UpperBind=255,int ch2LowerBind=0,
               int ch2UpperBind=255,int ch3LowerBind=0,int ch3UpperBind=255)
{
    
    
    Point pToGrowing;                       //待生长点位置
    int pGrowValue = 0;                             //待生长点灰度值
    Scalar pSrcValue = 0;                               //生长起点灰度值
    Scalar pCurValue = 0;                               //当前生长点灰度值
    Mat growImage = Mat::zeros(srcImage.size(), CV_8UC1);   //创建一个空白区域,填充为黑色
    //生长方向顺序数据
    int DIR[8][2] = {
    
    {
    
    -1,-1}, {
    
    0,-1}, {
    
    1,-1}, {
    
    1,0}, {
    
    1,1}, {
    
    0,1}, {
    
    -1,1}, {
    
    -1,0}};
    vector<Point> growPtVector;                     //生长点栈
    growPtVector.push_back(pt);                         //将生长点压入栈中
    growImage.at<uchar>(pt.y, pt.x) = 255;              //标记生长点
    pSrcValue = srcImage.at<Vec3b>(pt.y, pt.x);         //记录生长点的灰度值

    while (!growPtVector.empty())                       //生长栈不为空则生长
    {
    
    
        pt = growPtVector.back();                       //取出一个生长点
        growPtVector.pop_back();

        //分别对八个方向上的点进行生长
        for (int i = 0; i<9; ++i)
        {
    
    
            pToGrowing.x = pt.x + DIR[i][0];
            pToGrowing.y = pt.y + DIR[i][1];
            //检查是否是边缘点
            if (pToGrowing.x < 0 || pToGrowing.y < 0 ||
                    pToGrowing.x > (srcImage.cols-1) || (pToGrowing.y > srcImage.rows -1))
                continue;

            pGrowValue = growImage.at<uchar>(pToGrowing.y, pToGrowing.x);       //当前待生长点的灰度值
            pSrcValue = srcImage.at<Vec3b>(pt.y, pt.x);
            if (pGrowValue == 0)                    //如果标记点还没有被生长
            {
    
    
                pCurValue = srcImage.at<Vec3b>(pToGrowing.y, pToGrowing.x);
                if(pCurValue[0] <= ch1UpperBind && pCurValue[0] >= ch1LowerBind&&   //限制生长点的三通道上下界
                        pCurValue[1] <= ch2UpperBind && pCurValue[1] >= ch2LowerBind &&
                        pCurValue[2] <= ch3UpperBind && pCurValue[2] >= ch3LowerBind )
                {
    
    
                    if (abs(pSrcValue[0] - pCurValue[0]) < ch1Thres &&
                            abs(pSrcValue[1] - pCurValue[1]) < ch2Thres &&
                            abs(pSrcValue[2] - pCurValue[2]) < ch3Thres)                    //在阈值范围内则生长
                    {
    
    
                        growImage.at<uchar>(pToGrowing.y, pToGrowing.x) = 255;      //标记为白色
                        growPtVector.push_back(pToGrowing);                 //将下一个生长点压入栈中
                    }
                }
            }
        }
    }
    return growImage.clone();
}

参考

https://developer.aliyun.com/article/666729

猜你喜欢

转载自blog.csdn.net/weixin_42990464/article/details/130574502