it编程 > 平面设计 > Photoshop

深度剖析图像处理—边缘检测

220人参与 2024-08-01 Photoshop

什么是边缘检测

边缘类型

边缘检测是一种邻域运算,即一个像素是否是边缘点是由其所处的邻域决定的。在一定大小的邻域内,边缘分为阶跃边缘(step edge)和屋顶状边缘(roof edge)两种类型。下面以一维信号为例,分析这两种不同类型的边缘的导数特征

image-20240421200943004

求导与差分

在边缘检测中,导数的计算通常采用两种方法:

image-20240421202909028

image-20240421203632870

边缘强度与边缘方向

导数是有大小也有方向的,因此边缘也有强弱与方向,分别叫做边缘强度(edge intensity)和边缘方向(edge direction),边缘强度即边缘的幅值(magnitude)。用m(x,y)代表边缘的强度,θ(x,y)代表边缘的方向,有:

image-20240421204859267

实例

我们该如何提取这张图片的边缘呢?

test

首先当然是需要我们写一个函数来把24位彩色图像转化为8位灰度值图像

//24位彩色图像转8位灰度值
//rgbimage原始图像
//grayimage输出灰度图像
//width,height图片的宽和高
void converttogray(uint8_t* rgbimage, uint8_t* grayimage, int width, int height)
{
    for (int y = 0; y < height; y++) {
        for (int x = 0; x < width; x++) {
            // 获取当前像素的 rgb 分量
            uint8_t r = rgbimage[3 * (y * width + x) + 0];
            uint8_t g = rgbimage[3 * (y * width + x) + 1];
            uint8_t b = rgbimage[3 * (y * width + x) + 2];

            // 计算灰度值(常用的加权平均法)
            // 这里使用的加权系数是常见的:r: 0.299, g: 0.587, b: 0.114
            uint8_t gray = (uint8_t)(0.299 * r + 0.587 * g + 0.114 * b);

            // 将灰度值写入灰度图像数组
            grayimage[y * width + x] = gray;
        }
    }
}

我们来看看效果

image-20240422174937601

那接下来就需要用边缘检测来提取边缘了

一阶微分算子

梯度算子

梯度的本意是一个向量(矢量),表示某一函数在该点处的方向导数沿着该方向取得最大值,即函数在该点处沿着该方向(此梯度的方向)变化最快,变化率最大(为该梯度的模)。梯度的含义和边缘点是一致的,因此产生了边缘检测的梯度算子(gradient operator)

image-20240422172912964

写成模板的情况就是如下图所示:

image-20240422172958535

但是上面我们只是给出了像素(x,y)的边缘强度,称为梯度值;但是它是不是边缘点,还需要一定的约束条件,比如,设定当gradient(x,y)≥threshold时,像素(x,y)才是边缘点,threshold称为阈值。

image-20240422173629195

这就是用梯度算子计算的结果,我们上代码看看

void rmwgradientgryimg(uint8_t* pgryimg, int width, int height, uint8_t* pgrdimg)
{
    uint8_t* pgry, * pgrd;
    int dx, dy;
    int x, y;

    for (y = 0, pgry = pgryimg, pgrd = pgrdimg; y < height - 1; y++)
    {
        for (x = 0; x < width - 1; x++, pgry++)
        {
            dx = *pgry - *(pgry + 1);
            dy = *pgry - *(pgry + width);
            int gradient = (int)(sqrt(dx * dx * 1.0 + dy * dy));
            *pgrd++ = (gradient > 255) ? 255 : gradient;
        }
        *pgrd++ = 0; //尾列不做,边缘强度赋0
        pgry++;
    }
    memset(pgrd, 0, width); //尾行不做,边缘强度赋0
}

这里简单说一下这个函数

看看实际运行的效果是什么样子的

image-20240422180246550

效果并不是很好,我们把原始图像做一下图像增强(灰度值均衡化)试试

image-20240422181123926

换一种图像增强的方法,加一下反相试试

先简单写一下反相的函数

void invertimage(uint8_t *image, int width, int height) {
    for (int i = 0; i < width * height; i++) {
        image[i] = 255 - image[i];
    }
}

这次使用了线性拉伸并简单处理了一下参数

image-20240422181718341

效果还算不错,那如果加一下阈值呢?就像上面说的,我们重新写一下这个函数

//梯度算子加阈值
//pgryimg:输入的灰度图像数据指针。
//width:图像的宽度。
//height:图像的高度。
//pgrdimg:输出的梯度图像数据指针
void rmwgradientgryimgplus(uint8_t* pgryimg, int width, int height, uint8_t* pgrdimg, int threshold)
{
    uint8_t* pgry, * pgrd;
    int dx, dy;
    int x, y;

    for (y = 0, pgry = pgryimg, pgrd = pgrdimg; y < height - 1; y++)
    {
        for (x = 0; x < width - 1; x++, pgry++)
        {
            dx = *pgry - *(pgry + 1);
            dy = *pgry - *(pgry + width);
            int gradient = (int)(sqrt(dx * dx * 1.0 + dy * dy));
            *(pgrd++) = (gradient > threshold) ? min(255, gradient) : 0;
        }
        *(pgrd++) = 0; //尾列不做,边缘强度赋0
        pgry++;
    }
    memset(pgrd, 0, width); //尾行不做,边缘强度赋0
}

看一下效果

image-20240422182112492

画面确实变得更干净了


罗伯特算子

梯度算子的计算只涉及到了3个像素,只在水平和垂直方向做差分罗伯特算子(roberts operator)给出了一个4个像素之间进行运算的算子,分别在两个对角线方向做差分

image-20240422182433397

​ 由于对角线上2个像素之间的距离为√2,所以罗伯特算子的∆x和∆y采用对角线差分后,不再采用√(∆_x2+∆_y2 ),其描述见下式。罗伯特算子取∆x绝对值与∆y绝对值中的最大值。

image-20240422182518316

我们来看演示

image-20240422182547389

那么罗伯特算子好在哪呢?

罗伯特算子去掉了梯度算子的开方运算,计算复杂度也降低了不少。

void rmwrobertsgryimg(uint8_t *pgryimg, int width, int height, uint8_t *prbtimg)
{
    uint8_t *pgry, *prbt;
    int dx, dy;
    int x, y;

    for (y = 0, pgry = pgryimg, prbt = prbtimg; y < height - 1; y++)
    {
        for (x = 0; x < width - 1; x++, pgry++)
        {
            dx = *pgry - *(pgry + width + 1);
            dy = *(pgry + 1) - *(pgry + width);
            *prbt++ = (uint8_t)(dx > dy ? dx : dy); // 使用三目运算符选择较大的值
        }
        *prbt++ = 0; // 尾列不做, 边缘强度赋0
        pgry++;
    }
    memset(prbt, 0, width); // 尾行不做, 边缘强度赋0
}

我们来看结果

image-20240422192132132

为什么看起来的效果好像还没有梯度算子的结果好呢

是因为图像太复杂没有滤波,那么在边缘计算前能不能先滤波?

我们来看下一个算子

索贝尔算子

在一幅噪声较大的图像中,如果不进行图像平滑就进行边缘检测,必然会在边缘图像中产生噪声干扰

因此,索贝尔算子(sobel operator)中,在求∆x和∆y前,先进行滤波。在求∆x前,先执行如下图的所示的高斯均值滤波;在求∆y前,先执行如下图的所示的高斯均值滤波。

image-20240422184902956

另外,索贝尔算子进一步拉大进行差分的2个像素之间的距离,∆x和∆y采用如下的模板形式

image-20240422184942115

我们来看示意图

image-20240422185037120

在图像1中,红色方块代表当前像素(x,y),先执行图4-8(a)所示的高斯滤波,用d ̅、e ̅代表滤波后的值,则得到:

image-20240422185102632

执行∆x模板,则有∆x=d ̅-e ̅=(a+2d+f)-(c+2e+h)

image-20240422185448932

另外,索贝尔算子在对∆x和∆y的使用上,采用了它们的绝对值相加的形式

image-20240422185539273

看效果:

image-20240422185603424

对原始灰度图像执行索贝尔算子得到的结果,图中虚线框所示的边缘变成了双线宽

我们来写代码

//索贝尔算子
void rmwsobelgryimg(uint8_t* pgryimg, int width, int height, uint8_t* psbimg)
{
    uint8_t* pgry, * psb;
    int dx, dy;
    int x, y;

    memset(psbimg, 0, width); // 首行不做, 边缘强度赋0
    for (y = 1, pgry = pgryimg + width, psb = psbimg + width; y < height - 1; y++)
    {
        *psb++ = 0; // 首列不做, 边缘强度赋0
        pgry++;
        for (x = 1; x < width - 1; x++, pgry++)
        {
            // 求dx
            dx = *(pgry - 1 - width) + (*(pgry - 1) * 2) + *(pgry - 1 + width);
            dx -= *(pgry + 1 - width) + (*(pgry + 1) * 2) + *(pgry + 1 + width);
            // 求dy
            dy = *(pgry - width - 1) + (*(pgry - width) * 2) + *(pgry - width + 1);
            dy -= *(pgry + width - 1) + (*(pgry + width) * 2) + *(pgry + width + 1);
            // 结果
            *psb++ = (uint8_t)min(255, abs(dx) + abs(dy));
        }
        *psb++ = 0; // 尾列不做, 边缘强度赋0
        pgry++;
    }
    memset(psb, 0, width); // 尾行不做, 边缘强度赋0
}

image-20240422191511580

梯度算子、罗伯特算子、索贝尔算子的比较

以下从4个方面对梯度算子、罗伯特算子、索贝尔算子进行比较。

•1. 偏导数∆x和∆y的求取

梯度算子在3个像素之间进行运算,只在水平和垂直方向做差分,做差分的2个像素之间的距离为1。

罗伯特算子在4个像素之间进行运算,分别在两个对角线方向做差分,做差分的2个像素之间的距离为√2。

索贝尔算子在8个像素之间进行运算,只在水平和垂直方向做差分,做差分的2个像素之间的距离为2。

•2. 是否“先平滑后求导”

索贝尔算子在差分之前,进行了加权均值滤波对图像进行平滑(加权函数采用了高斯函数),因此索贝尔具有滤除噪声的效果。梯度算子和罗伯特算子都没有进行平滑。“先平滑后求导”是边缘检测的通用策略,一般在执行梯度算子和罗伯特算子前是需要使用另外的步骤做图像平滑,索贝尔算子则是把平滑写到了算子中。

•3. 边缘强度的大小

按照式(4-4)边缘强度的定义,梯度算子是严格遵守的,罗伯特算子是取∆x绝对值和∆y绝对值中的最大值,索贝尔算子是取∆x与∆y的绝对值之和。而且,索贝尔算子在高斯滤波后没有除以4,所以又相当于∆x、∆y放大了4倍。对于边缘强度,罗伯特算子、梯度算子、索贝尔算子之间的数值关系大致如下:

image-20240422195609623

•4. 邻域与边缘宽度

梯度算子、罗伯特算子的计算只涉及到了2行2列,所以它们得到的边缘宽度是1个像素;索贝尔算子涉及到了3行3列,所以它得到的边缘宽度是2个像素,边缘变成了双线宽。

方向模板

prewitt算子

prewitt算子设定了0°、45°、90°和135°,共计4种边缘方向;根据这4种边缘方向,分别设计了4个模板

image-20240422195755759

对每个像素分别计算这个4个模板的值,取绝对值最大者作为该像素的边缘强度prewitt(x,y)。同时该最大值对应模板的方向作为该像素的边缘方向(与边缘的走向相差90°,因为显然边缘走向的法线方向上的导数最大)。若把这些模板中为“0”点(空白处)的连成一条直线,可以发现这些模板强调了水平线、135°斜线、竖直线、45°斜线的检测。prewitt算子强调对直线的检测,对于上述走向的直线,总有一个模板的输出值最大

void rmwprewittgryimg(uint8_t *pgryimg, int width, int height, uint8_t *pprtimg)
{
    uint8_t *pgry, *pprt;
    int dx, dy, d45, d135, v1, v2;
    int x, y;

    memset(pprtimg, 0, width); // 首行不做, 边缘强度赋0
    for (y = 1, pgry = pgryimg + width, pprt = pprtimg + width; y < height - 1; y++)
    {
        *pprt++ = 0; // 首列不做, 边缘强度赋0
        pgry++;
        for (x = 1; x < width - 1; x++, pgry++)
        {
            // 求dx
            dx = *(pgry - 1 - width) + *(pgry - 1) + *(pgry - 1 + width);
            dx -= *(pgry + 1 - width) + *(pgry + 1) + *(pgry + 1 + width);
            // 求dy
            dy = *(pgry - width - 1) + *(pgry - width) + *(pgry - width + 1);
            dy -= *(pgry + width - 1) + *(pgry + width) + *(pgry + width + 1);
            // 求45度
            d45 = *(pgry - width - 1) + *(pgry - width) + *(pgry - 1);
            d45 -= *(pgry + width + 1) + *(pgry + width) + *(pgry + 1);
            // 求135度
            d135 = *(pgry - width) + *(pgry - width + 1) + *(pgry + 1);
            d135 -= *(pgry + width - 1) + *(pgry + width) + *(pgry - 1);
            // 结果
            v1 = abs(dx) > abs(dy) ? abs(dx) : abs(dy);
            v2 = abs(d45) > abs(d135) ? abs(d45) : abs(d135);
            *pprt++ = (uint8_t)((v1 > v2) ? ((v1 > 255) ? 255 : v1) : ((v2 > 255) ? 255 : v2));
        }
        *pprt++ = 0; // 尾列不做, 边缘强度赋0
        pgry++;
    }
    memset(pprt, 0, width); // 尾行不做, 边缘强度赋0
}

我们看一下实现效果

image-20240422200701865

robinson算子

robinson算子除(a)外的7个模板,都是由其上个模板顺时针旋转1个像素得到的。若是把模板中的负数值合并成一个区域,可以看出该算子强调了对角点的检测;对于各种形状的角点,总有一个模板的输出值最大。

image-20240422200914550

有兴趣大家可以查一下啊,这里就不再过多解释

二阶微分算子

一阶微分算子能够得到边缘强度,但是需要再加上一定的条件约束,比如设置个阈值,才能判定一个像素是不是边缘点。

通过对边缘类型及其导数的分析可知,阶跃边缘的导数特征除了“一阶导数取极值”外,还有“二阶导数过零点”。因此可以采用二阶导数,利用过零点得到边缘点,这样就不需要其他的条件了。

拉普拉斯算子

拉普拉斯算子(laplacian operator)是近似给出二阶导数的流行方法,其使用3×3的邻域,给出了4-邻接和8-邻接的邻域的2种模板

image-20240422201917270

对如图所示的原始灰度图像执行4邻域拉普拉斯算子,得到的结果如图所示,图中虚线框所示的位置上发生了过零点(导数由负数变到了正数),此处即边缘。

image-20240422202053683

​ 在拉普拉斯算子的结果图像中,可以发现过零点位置刚好就是边缘的位置。由于过零点是在像素之间,不在整数坐标上,所以在提取边缘点时,往往采取下面的策略:

当一个像素的二阶导数大于0,其邻域内有像素的二阶导数小于0或等于0,则该像素被标记为边缘点。

沈俊算子

唯一一个以国人命名的算子

沈教授在阶跃边缘和可加白噪声模型下,就信噪比最大准则,证明了图像平滑的最佳滤波器是对称的指数函数,形式如下:

image-20240422202508333

显然,当a_0越大时,c2就越小,t(j,i)就越陡越窄,相当于滤波邻域就越小,压制噪声的能力就弱,图像模糊程度就越小,边缘定位的精度就越高。

在算子实现上,沈教授对图像分别按行、按列各进行两次先正方向再反方向的递推滤波实现(|j|、|i|的优点),等价于用上述指数函数进行图像滤波;证明了滤波结果减去原始灰度值得到的差值乘以2c1〖ln〗^c2,约等于其二阶导数的值。沈俊算子的实现过程如下:

沈俊算子的实现过程
step.1 对每行从左向右进行:
𝑔1(0,𝑦)=𝑔(0,𝑦),
𝑔1(𝑥,𝑦)=𝑔1(𝑥−1,𝑦)+𝑎_0×(𝑔(𝑥,𝑦)−𝑔1(𝑥−1,𝑦)),𝑥=1,2,⋯𝑤𝑖𝑑𝑡ℎ−1。
step.2 对每行从右向左进行:
𝑔2(𝑤𝑖𝑑𝑡ℎ−1,𝑦)=𝑔1(𝑤𝑖𝑑𝑡ℎ−1,𝑦),
𝑔2(𝑥,𝑦)=𝑔2(𝑥+1,𝑦)+𝑎_0×(𝑔1(𝑥,𝑦)−𝑔2(𝑥+1,𝑦)),𝑥=𝑤𝑖𝑑𝑡ℎ−2,, 1,0。
step.3 对每列从上向下进行:
𝑔3(𝑥,0)=𝑔2(𝑥,0),
𝑔3(𝑥,𝑦)=𝑔3(𝑥,𝑦−1)+𝑎_0×(𝑔2(𝑥,𝑦)−𝑔3(𝑥,𝑦−1)),𝑦=1,2,⋯ℎ𝑒𝑖𝑔ℎ𝑡−1。
step.4 对每列从下向上进行:
𝑔4(𝑥,ℎ𝑖𝑔ℎ𝑡−1)=𝑔3(𝑥,ℎ𝑖𝑔ℎ𝑡−1),
𝑔4(𝑥,𝑦)=𝑔4(𝑥,𝑦+1)+𝑎_0×(𝑔3(𝑥,𝑦)−𝑔4(𝑥,𝑦+1)),𝑦=ℎ𝑒𝑖𝑔ℎ𝑡−2,, 1,0。
step.5 对每个像素(𝑥,𝑦)执行𝑆𝐽(𝑥,𝑦)=𝑔4(𝑥,𝑦)−𝑔(𝑥,𝑦),得到二阶导数𝑆𝐽(𝑥,𝑦)。
step.6 对每个像素𝑆𝐽(𝑥,𝑦)进行过零点检测得到边缘点。

我们来看程序

//沈俊算子
//pgryimg 和 ptmpimg 是指向 uint8_t 类型的指针,它们分别指向原始灰度图像数据和辅助图像数据。
//width 和 height 是整型参数,表示图像的宽度和高度。
//a0 是双精度浮点型参数,表示滤波系数。
//psjimg 是指向 uint8_t 类型的指针,它指向了输出的图像数据。
void rmwshenjungryimg(uint8_t* pgryimg,uint8_t* ptmpimg, int width, int height, double a0, uint8_t* psjimg)
{
    uint8_t* pgry, * pcur, * psj, * pend;
    int lut[512], * alut; // a0查找表
    int x, y, pre, dif;

    // step 1: 初始化查找表
    a0 = (a0 < 0.01) ? 0.01 : ((a0 > 0.99) ? 0.99 : a0); // 安全性检查
    // a0查找表, 进行了四舍五入
    alut = lut + 256;
    for (alut[0] = 0, dif = 1; dif < 256; dif++)
    {
        alut[dif] = (int)(dif * a0 + 0.5);
        alut[-dif] = (int)(-dif * a0 - 0.5);
    }

    // step 2: 递推实现指数滤波
    // 按行滤波
    for (y = 0, pgry = pgryimg, pcur = ptmpimg; y < height; y++)
    {
        // 从左向右: p1(y,x) = p1(y,x-1) + a * [p(y,x) - p1(y,x-1)]
        *(pcur++) = pre = *(pgry++);
        for (x = 1; x < width; x++, pgry++)
            *(pcur++) = pre = pre + alut[*pgry - pre];
        pcur--; // 回到行尾
        // 从右向左: p2(y,x) = p2(y,x+1) - a * [p1(y,x) - p2(y,x+1)]
        for (x = width - 2, pcur = pcur - 1; x >= 0; x--)
            *(pcur--) = pre = pre + alut[*pcur - pre];
        pcur += (width + 1); // 回到下一行的开始
    }
    // 按列滤波
    for (x = 0, pcur = ptmpimg; x < width; x++, pcur = ptmpimg + x)
    {
        // 从上向下: p3(y,x) = p3(y-1,x) + a * [p2(y,x) - p3(y-1,x)]
        pre = *pcur;
        for (y = 1, pcur += width; y < height; y++, pcur += width)
            *pcur = pre = pre + alut[*pcur - pre];
        pcur -= width; // 回到列尾
        // 从下向上: p4(i,j) = p4(i+1,j) + a * [p3(i,j) - p4(i+1,j)]
        for (y = height - 2, pcur -= width; y >= 0; y--, pcur -= width)
            *pcur = pre = pre + alut[*pcur - pre];
    }

    // step 3: 正导数=1,负导数为0,0必须也是0
    pend = ptmpimg + width * height;
    for (pcur = ptmpimg, pgry = pgryimg; pcur < pend; pgry++)
    {
        *(pcur++) = (*pcur > *pgry);
    }

    // step 4: 过零点检测
    memset(psjimg, 0, width * height); // 边缘强度赋0
    psj = psjimg + width;
    pcur = ptmpimg + width; // 首行不做 
    for (y = 1; y < height - 1; y++)
    {
        psj++; pcur++;  // 首列不做
        for (x = 1; x < width - 1; x++, pgry++, pcur++, psj++)
        {
            if (*pcur) // 正导数
            {
                // 下面使用4邻域, 边缘为8连通, 不能保证4连通; 使用8邻域才能保证边缘4连通
                if ((!*(pcur - 1)) || // 左, 必须<=0, 不能<0
                    (!*(pcur + 1)) || // 右, 必须<=0, 不能<0
                    (!*(pcur - width)) || // 上, 必须<=0, 不能<0
                    (!*(pcur + width)))   // 下, 必须<=0, 不能<0
                {
                    *psj = 255; // 周围有导数小于等于0
                }
            }
        }
        psj++; pcur++;  // 尾列不做
    }
}

当滤波系数为0.1时,我们来看效果

image-20240422204121096

测试

那么该如何更好的得到我们原始图像的边缘呢

可能一个边缘处理都没有办法得到很好的效果,我们可以结合多个处理办法

image-20240422204713809

我们可以看到效果比较好的是索贝尔算子,那么我们结合一下

//沈俊算子加索贝尔算子
//pgryimg:指向原始灰度图像数据的指针
//ptmpimg:指向辅助图像数据的指针
//width:图像的宽度
//height:图像的高度
//a0:这是沈俊算子的参数,用于控制边缘检测的灵敏度。
//grdthre:这是sobel算子的梯度阈值
//pedgeimg:最终边缘图像数据的指针
void rmwextractriceedge(uint8_t* pgryimg,uint8_t* ptmpimg,int width,int height,double a0, int grdthre, uint8_t* pedgeimg)
{
    // step.1------------沈俊算子-----------------------//
    rmwshenjungryimg(pgryimg, ptmpimg, width, height, a0, pedgeimg);
    // step.2------------sobel算子----------------------//
    rmwsobelgryimg(pgryimg, width, height, ptmpimg);
    // step.3------------二者融合-----------------------//
    for (int i = 0; i < width * height; i++)
    {
        *(pedgeimg + i) = (pedgeimg[i] && (ptmpimg[i] > grdthre)) * 255;
    }
    // step.4------------结束---------------------------//
    return;
}

image-20240422211324490

我们换一张图片试一下

image-20240422212038348

总结

那么到目前为止,我们已经可以成功的进行边缘检测

接下来就是边缘增强,边缘分割等相关内容,会尽快更新

源码

idp.h

#pragma once

#include <stdio.h>
#include <stdint.h>
#include <stdlib.h>
#include <math.h>
#include <nmmintrin.h>
 
uint8_t* readgrayscalebmp(const char* filename, int* width, int* height);//读取8位灰度图片
void savegrayscalebmp(const char* filename, const uint8_t* imagedata, int width, int height);// 将8位灰度图像数据保存为bmp文件
uint8_t* readcolorbmp(const char* filename, int* width, int* height);//读取24位彩色图像的bmp文件
void savecolorbmp(const char* filename, const uint8_t* imagedata, int width, int height);//将24位彩色图像数据保存为bmp文件
void converttogray(uint8_t* rgbimage, uint8_t* grayimage, int width, int height);//24位彩色图像转8位灰度值
void linearstretchdemo(uint8_t* pgryimg, int width, int height, double k, double b);//灰度线性拉伸
void gethistogram(uint8_t* pimg, int width, int height, int* histogram);//统计图像灰度值
void getbrightcontrast(int* histogram, double* bright, double* contrast);//亮度和对比度
void rmwhistogramequalize(uint8_t* pgryimg, int width, int height);//直方图均衡化
void rmwlogtransform(uint8_t* pgryimg, int width, int height);//对数变换
void rmwavrfilterbysumcol(uint8_t* pgryimg, int width, int height, int m, int n, uint8_t* presimg);//基于列积分的快速均值滤波
void rmwdosumgryimg(uint8_t* pgryimg, int width, int height, int* psumimg);//基于列积分的积分图实现
void rmwdosumgryimg_sse(uint8_t* pgryimg, int width, int height, int* psumimg);//基于sse的积分图实现
void rmwavrfilterbysumimg(int* psumimg, int width, int height, int m, int n, uint8_t* presimg);//基于积分图的快速均值滤波  
void getmediangry(int* histogram, int n, int* medgry);//求灰度值中值
double rmwmedianfilter(uint8_t* pgryimg, int width, int height, int m, int n, uint8_t* presimg);//中值滤波
void rmwbinimgfilter(uint8_t* pbinimg, int width, int height, int m, int n, double threshold, uint8_t* presimg);//二值滤波
void rmwgradientgryimg(uint8_t* pgryimg, int width, int height, uint8_t* pgrdimg);//梯度算子
void rmwgradientgryimgplus(uint8_t* pgryimg, int width, int height, uint8_t* pgrdimg, int threshold);//梯度算子加阈值
void invertimage(uint8_t* image, int width, int height);//反相
void rmwrobertsgryimg(uint8_t* pgryimg, int width, int height, uint8_t* prbtimg);//罗伯特算子
void rmwsobelgryimg(uint8_t* pgryimg, int width, int height, uint8_t* psbimg);//索贝尔算子
void rmwprewittgryimg(uint8_t* pgryimg, int width, int height, uint8_t* pprtimg); //prewitt算子
void rmwshenjungryimg(uint8_t* pgryimg, uint8_t* ptmpimg, int width, int height, double a0, uint8_t* psjimg);//沈俊算子
void rmwextractriceedge(uint8_t* pgryimg, uint8_t* ptmpimg, int width, int height, double a0, int grdthre, uint8_t* pedgeimg);//索贝尔+沈俊算子

idp.c

#define _crt_secure_no_warnings 1

#include "idp.h"

//读取8位灰度图片
//filename:字符数组的指针,用于指定要保存的图像文件的名称或路径。
//imagedata:无符号 8 位整型数据的指针,代表要保存的图像数据。
//width:图像的宽度。
//height:图像的高度。
uint8_t* readgrayscalebmp(const char* filename, int* width, int* height) 
{
    file* file = fopen(filename, "rb");
    if (!file) {
        fprintf(stderr, "error opening file %s\n", filename);
        return null;
    }

    // 读取bmp文件头部信息
    uint8_t bmpheader[54];
    fread(bmpheader, 1, 54, file);

    // 从文件头部提取图像宽度和高度信息
    *width = *(int*)&bmpheader[18];
    *height = *(int*)&bmpheader[22];

    // 分配存储图像数据的内存
    uint8_t* imagedata = (uint8_t*)malloc(*width * *height);
    if (!imagedata) {
        fprintf(stderr, "内存分配失败\n");
        fclose(file);
        return null;
    }

    // 计算调色板的大小
    int palettesize = *(int*)&bmpheader[46];
    if (palettesize == 0)
        palettesize = 256;

    // 读取调色板数据
    uint8_t palette[1024];
    fread(palette, 1, palettesize * 4, file);

    // 读取图像数据
    fseek(file, *(int*)&bmpheader[10], seek_set);
    fread(imagedata, 1, *width * *height, file);

    fclose(file);

    return imagedata;
}

// 将8位灰度图像数据保存为bmp文件
//filename:字符数组的指针,用于指定要保存的图像文件的名称或路径。
//imagedata:无符号 8 位整型数据的指针,代表要保存的图像数据。
//width:图像的宽度。
//height:图像的高度。
void savegrayscalebmp(const char* filename, const uint8_t* imagedata, int width, int height) 
{
    file* file = fopen(filename, "wb");
    if (!file) {
        fprintf(stderr, "error creating file %s\n", filename);
        return;
    }

    // bmp文件头部信息
    uint8_t bmpheader[54] = {
        0x42, 0x4d,             // 文件类型标识 "bm"
        0x36, 0x00, 0x0c, 0x00, // 文件大小(以字节为单位,此处假设图像数据大小不超过4gb)
        0x00, 0x00,             // 保留字段
        0x00, 0x00,             // 保留字段
        0x36, 0x00, 0x00, 0x00, // 位图数据偏移(以字节为单位)
        0x28, 0x00, 0x00, 0x00, // 位图信息头大小(40字节)
        0x00, 0x00, 0x00, 0x00, // 图像宽度
        0x00, 0x00, 0x00, 0x00, // 图像高度
        0x01, 0x00,             // 目标设备的级别(此处为1,不压缩)
        0x08, 0x00,             // 每个像素的位数(8位)
        0x00, 0x00, 0x00, 0x00, // 压缩类型(此处为不压缩)
        0x00, 0x00, 0x00, 0x00, // 图像数据大小(以字节为单位,此处为0,表示不压缩)
        0x00, 0x00, 0x00, 0x00, // 水平分辨率(像素/米,此处为0,表示未知)
        0x00, 0x00, 0x00, 0x00, // 垂直分辨率(像素/米,此处为0,表示未知)
        0x00, 0x00, 0x00, 0x00, // 使用的颜色索引数(0表示使用所有调色板项)
        0x00, 0x00, 0x00, 0x00  // 重要的颜色索引数(0表示所有颜色都重要)
    };

    // 更新bmp文件头部信息中的宽度和高度
    *(int*)&bmpheader[18] = width;
    *(int*)&bmpheader[22] = height;

    // 写入bmp文件头部信息
    fwrite(bmpheader, 1, 54, file);

    // 写入调色板数据
    for (int i = 0; i < 256; i++) {
        fputc(i, file);  // 蓝色分量
        fputc(i, file);  // 绿色分量
        fputc(i, file);  // 红色分量
        fputc(0, file);  // 保留字节
    }

    // 写入图像数据
    fwrite(imagedata, 1, width * height, file);

    fclose(file);
}

// 读取24位彩色图像的bmp文件
//filename:字符数组的指针,用于指定要读取的 bmp 格式图像文件的名称或路径。
//width:整型变量的指针,用于存储读取的图像的宽度。
//height:整型变量的指针,用于存储读取的图像的高度。
uint8_t* readcolorbmp(const char* filename, int* width, int* height) 
{
    file* file = fopen(filename, "rb");
    if (!file) {
        fprintf(stderr, "error opening file %s\n", filename);
        return null;
    }

    // 读取bmp文件头部信息
    uint8_t bmpheader[54];
    fread(bmpheader, 1, 54, file);

    // 从文件头部提取图像宽度和高度信息
    *width = *(int*)&bmpheader[18];
    *height = *(int*)&bmpheader[22];

    // 分配存储图像数据的内存
    uint8_t* imagedata = (uint8_t*)malloc(*width * *height * 3);
    if (!imagedata) {
        fprintf(stderr, "memory allocation failed\n");
        fclose(file);
        return null;
    }

    // 读取图像数据
    fseek(file, *(int*)&bmpheader[10], seek_set);
    fread(imagedata, 1, *width * *height * 3, file);

    fclose(file);

    return imagedata;
}

//将24位彩色图像数据保存为bmp文件
//filename:字符数组的指针,用于指定要保存的图像文件的名称或路径。
//imagedata:无符号 8 位整型数据的指针,代表要保存的图像数据。
//width:图像的宽度。
//height:图像的高度。
void savecolorbmp(const char* filename, const uint8_t* imagedata, int width, int height) 
{
    file* file = fopen(filename, "wb");
    if (!file) {
        fprintf(stderr, "error creating file %s\n", filename);
        return;
    }

    // bmp文件头部信息
    uint8_t bmpheader[54] = {
        0x42, 0x4d,             // 文件类型标识 "bm"
        0x00, 0x00, 0x00, 0x00, // 文件大小(占位,稍后计算)
        0x00, 0x00,             // 保留字段
        0x00, 0x00,             // 保留字段
        0x36, 0x00, 0x00, 0x00, // 位图数据偏移(以字节为单位)
        0x28, 0x00, 0x00, 0x00, // 位图信息头大小(40字节)
        0x00, 0x00, 0x00, 0x00, // 图像宽度
        0x00, 0x00, 0x00, 0x00, // 图像高度
        0x01, 0x00,             // 目标设备的级别(此处为1,不压缩)
        0x18, 0x00,             // 每个像素的位数(24位)
        0x00, 0x00, 0x00, 0x00, // 压缩类型(此处为不压缩)
        0x00, 0x00, 0x00, 0x00, // 图像数据大小(占位,稍后计算)
        0x00, 0x00, 0x00, 0x00, // 水平分辨率(像素/米,此处为0,表示未知)
        0x00, 0x00, 0x00, 0x00, // 垂直分辨率(像素/米,此处为0,表示未知)
        0x00, 0x00, 0x00, 0x00, // 使用的颜色索引数(0表示使用所有调色板项)
        0x00, 0x00, 0x00, 0x00  // 重要的颜色索引数(0表示所有颜色都重要)
    };

    // 更新bmp文件头部信息中的宽度和高度
    *(int*)&bmpheader[18] = width;
    *(int*)&bmpheader[22] = height;

    // 计算图像数据大小
    uint32_t imagedatasize = width * height * 3 + 54; // 加上文件头部大小
    bmpheader[2] = (uint8_t)(imagedatasize & 0xff);
    bmpheader[3] = (uint8_t)((imagedatasize >> 8) & 0xff);
    bmpheader[4] = (uint8_t)((imagedatasize >> 16) & 0xff);
    bmpheader[5] = (uint8_t)((imagedatasize >> 24) & 0xff);

    // 写入bmp文件头部信息
    fwrite(bmpheader, 1, 54, file);

    // 写入图像数据
    fwrite(imagedata, width * height * 3, 1, file);

    fclose(file);
}

//24位彩色图像转8位灰度值
//rgbimage原始图像
//grayimage输出灰度图像
//width,height图片的宽和高
void converttogray(uint8_t* rgbimage, uint8_t* grayimage, int width, int height)
{
    for (int y = 0; y < height; y++) {
        for (int x = 0; x < width; x++) {
            // 获取当前像素的 rgb 分量
            uint8_t r = rgbimage[3 * (y * width + x) + 0];
            uint8_t g = rgbimage[3 * (y * width + x) + 1];
            uint8_t b = rgbimage[3 * (y * width + x) + 2];

            // 计算灰度值(常用的加权平均法)
            // 这里使用的加权系数是常见的:r: 0.299, g: 0.587, b: 0.114
            uint8_t gray = (uint8_t)(0.299 * r + 0.587 * g + 0.114 * b);

            // 将灰度值写入灰度图像数组
            grayimage[y * width + x] = gray;
        }
    }
}

//灰度线性拉伸
//pgryimg:灰度图像数据的指针。
//width:图像的宽度。
//height:图像的高度。
//k:线性拉伸的斜率。它控制着拉伸的速率或程度。当(k) 大于 1 时,图像的对比度增加;当(k) 小于 1 时,对比度降低。
//b:线性拉伸的偏移。它控制着拉伸后灰度值的起始位置。当(b) 大于 0 时,图像的整体亮度增加;当(b) 小于 0 时,整体亮度减小。
void linearstretchdemo(uint8_t* pgryimg, int width, int height, double k, double b)
{
    uint8_t* pcur, * pend;
    int lut[256];    //因为只有[0,255]共256个灰度值

    //step1. 生成查找表
    for (int g = 0; g < 256; g++)
    {
        lut[g] = max(0, min(255, k * g + b));
    }

    //step2. 进行变换
    for (pcur = pgryimg, pend = pgryimg + width * height; pcur < pend; pcur++)
    {
        *pcur = lut[*pcur];
    }
    //step3. 结束
    return;
}

//统计图像灰度值
//pimg:灰度图像数据的指针。
//width:图像的宽度。
//height:图像的高度。
//* histogram:数组首元素地址,需要一个能储存256个变量的整型数组
void gethistogram(uint8_t* pimg, int width, int height, int* histogram)
{
    uint8_t* pcur;
    uint8_t* pend = pimg + width * height;

    // 初始化直方图数组
    memset(histogram, 0, sizeof(int) * 256);

    // 直方图统计
    for (pcur = pimg; pcur < pend;)
    {
        histogram[*pcur]++;
        pcur++;
    }

    // 函数结束
    return;
}

//亮度和对比度
//储存histogram灰度直方图的指针
//接收亮度的变量地址
//接收对比度的变量地址
void getbrightcontrast(int* histogram, double* bright, double* contrast)
{
    int g;
    double sum, num; //书上说图像很亮时,int有可能会溢出,所以我这里直接用double
    double fsum;

    //step.1 求亮度
    for (sum = num = 0, g = 0; g < 256; g++)
    {
        sum += histogram[g] * g;
        num += histogram[g];
    }
    *bright = sum * 1.0 / num;

    //step.2 求对比度
    for (fsum = 0.0, g = 0; g < 256; g++)
    {
        fsum += histogram[g] * (g - *bright) * (g - *bright);
    }
    *contrast = sqrt(fsum / (num - 1)); //即std dev

    //step.3 结束
    return;
}

//pgryimg:灰度图像数据的指针。
//width:图像的宽度。
//height:图像的高度。
void rmwhistogramequalize(uint8_t* pgryimg, int width, int height)
{
    uint8_t* pcur, * pend = pgryimg + width * height; // 指针变量,指向当前像素和图像末尾
    int histogram[256], lut[256], a, g; // 直方图数组、查找表数组、累积直方图、灰度级

    // step.1-------------求直方图--------------------------//
    memset(histogram, 0, sizeof(int) * 256); // 初始化直方图数组为0
    for (pcur = pgryimg; pcur < pend;)
        histogram[*(pcur++)]++; // 统计每个灰度级出现的频率

    // step.2-------------求lut[g]-------------------------//
    a = histogram[0]; // 初始化累积直方图的值为第一个灰度级的频率
    lut[0] = 255 * a / (width * height); // 计算第一个灰度级对应的均衡化后的灰度值
    for (g = 1; g < 256; g++) {
        a += histogram[g]; // 更新累积直方图的值
        lut[g] = 255 * a / (width * height); // 计算当前灰度级对应的均衡化后的灰度值
    }

    // step.3-------------查表------------------------------//
    for (pcur = pgryimg; pcur < pend;)
        *(pcur++) = lut[*pcur]; // 使用查找表对每个像素进行灰度映射

    // step.4-------------结束------------------------------//
    return;
}

//对数变换
//pgryimg:灰度图像数据的指针。
//width:图像的宽度。
//height:图像的高度。
void rmwlogtransform(uint8_t* pgryimg, int width, int height)
{
    uint8_t* pcur, * pend = pgryimg + width * height; // 指向灰度图像数据的当前指针和结束指针
    int histogram[256], lut[256], gmax, g; // 声明直方图数组、查找表数组、最大灰度值、当前灰度值
    double c; // 声明常数c

    // step.1-------------求直方图--------------------------//
    memset(histogram, 0, sizeof(int) * 256); // 初始化直方图数组为0
    for (pcur = pgryimg; pcur < pend;)
        histogram[*(pcur++)]++; // 遍历图像数据,统计每个灰度级的像素数量

    // step.2-------------最大值---------------------------//
    for (gmax = 255; gmax >= 0; gmax++)
        if (histogram[gmax]) break; // 从最大灰度级开始向低灰度级搜索,找到第一个非零灰度级,即最大灰度值

    // step.3-------------求lut[g]-------------------------//
    c = 255.0 / log(1 + gmax); // 计算常数c
    for (g = 0; g < 256; g++)
    {
        lut[g] = (int)(c * log(1 + g)); // 根据对数变换公式计算查找表中每个灰度级的映射值
    }

    // step.4-------------查表------------------------------//
    for (pcur = pgryimg; pcur < pend;)
        *(pcur++) = lut[*pcur]; // 使用查找表将图像数据进行对数变换

    // step.5-------------结束------------------------------//
    return; // 函数结束
}

//基于列积分的快速均值滤波
//原始灰度图像
//图像的宽度和高度
//滤波邻域:m列n行
//结果图像
void rmwavrfilterbysumcol(uint8_t* pgryimg,int width, int height,int m, int n,uint8_t* presimg) 
{
    uint8_t* padd, * pdel, * pres;
    int halfx, halfy;
    int x, y;
    int sum, c;
    int sumcol[4096]; // 约定图像宽度不大于4096

    // step.1------------初始化--------------------------//
    m = m / 2 * 2 + 1; // 奇数化
    n = n / 2 * 2 + 1; // 奇数化
    halfx = m / 2; // 滤波器的半径x
    halfy = n / 2; // 滤波器的半径y
    c = (1 << 23) / (m * n); // 乘法因子
    memset(sumcol, 0, sizeof(int) * width);
    for (y = 0, padd = pgryimg; y < n; y++) {
        for (x = 0; x < width; x++) sumcol[x] += *(padd++);
    }
    // step.2------------滤波----------------------------//
    for (y = halfy, pres = presimg + y * width, pdel = pgryimg; y < height - halfy; y++) {
        // 初值
        for (sum = 0, x = 0; x < m; x++) sum += sumcol[x];
        // 滤波
        pres += halfx; // 跳过左侧
        for (x = halfx; x < width - halfx; x++) {
            // 求灰度均值
            // *(pres++)=sum/(n*m);
            *(pres++) = (sum * c) >> 23; // 用整数乘法和移位代替除法
            // 换列,更新灰度和
            sum -= sumcol[x - halfx]; // 减左边列
            sum += sumcol[x + halfx + 1]; // 加右边列
        }
        pres += halfx; // 跳过右侧
        // 换行,更新sumcol
        for (x = 0; x < width; x++) {
            sumcol[x] -= *(pdel++); // 减上一行
            sumcol[x] += *(padd++); // 加下一行
        }
    }
    // step.3------------返回----------------------------//
    return;
}

//基于列积分的积分图实现
//pgryimg, // 原始灰度图像
//width,       // 图像的宽度 
//height,      // 图像的高度
//psumimg     // 计算得到的积分图
void rmwdosumgryimg(uint8_t* pgryimg,int width,int height, int* psumimg)
{
    uint8_t* pgry;
    int* pres;
    int x, y;
    int sumcol[4096]; // 约定图像宽度不大于4096

    memset(sumcol, 0, sizeof(int) * width);
    for (y = 0, pgry = pgryimg, pres = psumimg; y < height; y++)
    {
        // 最左侧像素的特别处理
        sumcol[0] += *(pgry++);
        *(pres++) = sumcol[0];
        // 正常处理
        for (x = 1; x < width; x++)
        {
            sumcol[x] += *(pgry++);       // 更新列积分
            int temp = *(pres - 1);
            *(pres++) = temp + sumcol[x];
        }
    }
    return;
}

//基于sse的积分图实现
//pgryimg原始灰度图像
//width图像的宽度,必须是4的倍数
//height图像的高度
//psumimg计算得到的积分图
void rmwdosumgryimg_sse(uint8_t* pgryimg,int width,int height,int* psumimg)
{
    int sumcol[4096]; //约定图像宽度不大于4096
    __m128i* psumsse, a;
    uint8_t* pgry;
    int* pres;
    int x, y;

    memset(sumcol, 0, sizeof(int) * width);
    for (y = 0, pgry = pgryimg, pres = psumimg; y < height; y++)
    {
        // 0:需要特别处理
        sumcol[0] += *(pgry++);
        *(pres++) = sumcol[0];
        // 1
        sumcol[1] += *(pgry++);
        *(pres++) = *(pres - 1) + sumcol[1];
        // 2
        sumcol[2] += *(pgry++);
        *(pres++) = *(pres - 1) + sumcol[2];
        // 3
        sumcol[3] += *(pgry++);
        *(pres++) = *(pres - 1) + sumcol[3];
        // [4...width-1]
        for (x = 4, psumsse = (__m128i*)(sumcol + 4); x < width; x += 4, pgry += 4)
        {
            // 把变量的低32位(有4个8位整数组成)转换成32位的整数
            a = _mm_cvtepu8_epi32(_mm_loadl_epi64((__m128i*)pgry));
            // 4个32位的整数相加
            *(psumsse++) = _mm_add_epi32(*psumsse, a);
            // 递推
            *(pres++) = *(pres - 1) + sumcol[x + 0];
            *(pres++) = *(pres - 1) + sumcol[x + 1];
            *(pres++) = *(pres - 1) + sumcol[x + 2];
            *(pres++) = *(pres - 1) + sumcol[x + 3];
        }
    }
    return;
}

//基于积分图的快速均值滤波
//psumimg计算得到的积分图
//width,height,图像的宽度和高度
//m, n,滤波邻域:m列n行
//presimg 结果图像
void rmwavrfilterbysumimg(int* psumimg,int width, int height,int m, int n,uint8_t* presimg)
{
    // 没有对边界上邻域不完整的像素进行处理,可以采用变窗口的策略
    int* py1, * py2;
    uint8_t* pres;
    int halfx, halfy;
    int y, x1, x2;
    int sum, c;

    // step.1------------初始化--------------------------//
    m = m / 2 * 2 + 1; // 奇数化
    n = n / 2 * 2 + 1; // 奇数化
    halfx = m / 2;      // 滤波器的半径x
    halfy = n / 2;      // 滤波器的半径y
    c = (1 << 23) / (m * n); // 乘法因子
    // step.2------------滤波----------------------------//
    for (y = halfy + 1, pres = presimg + y * width, py1 = psumimg, py2 = psumimg + n * width;
        y < height - halfy;
        y++, py1 += width, py2 += width)
    {
        pres += halfx + 1; // 跳过左侧
        for (x1 = 0, x2 = m; x2 < width; x1++, x2++) // 可以简化如此,但不太容易读
        {
            sum = *(py2 + x2) - *(py2 + x1) - *(py1 + x2) + *(py1 + x1);
            *(pres++) = (uint8_t)((sum * c) >> 23); // 用整数乘法和移位代替除法
        }
        pres += halfx; // 跳过右侧
    }
    // step.3------------返回----------------------------//
    return;
}

void getmediangry(int* histogram, int n, int* medgry)
{
    int g;
    int num;

    // step.1-------------求灰度中值------------------------//
    num = 0;
    for (g = 0; g < 256; g++)
    {
        num += histogram[g];
        if (2 * num > n) break;  //num>n/2
    }
    *medgry = g;
    // step.2-------------结束------------------------------//
    return;
}

//中值滤波
//pgryimg:指向待处理灰度图像数据的指针。
//width、height:表示图像的宽度和高度。
//m、n:分别表示中值滤波器的水平和垂直邻域大小(以像素为单位)。
//presimg:指向存储结果图像数据的指针。
double rmwmedianfilter(uint8_t* pgryimg, int width, int height, int m, int n, uint8_t* presimg) 
{
    uint8_t* pcur, * pres;
    int halfx, halfy, x, y, i, j, y1, y2;
    int histogram[256];
    int wsize, j1, j2;
    int num, med, v;
    int dbgcmptimes = 0; // 搜索中值所需比较次数的调试

    m = m / 2 * 2 + 1; // 奇数化
    n = n / 2 * 2 + 1; // 奇数化
    halfx = m / 2;      // x半径
    halfy = n / 2;      // y半径
    wsize = (halfx * 2 + 1) * (halfy * 2 + 1); // 邻域内像素总个数

    for (y = halfy, pres = presimg + y * width; y < height - halfy; y++) {
        // step.1----初始化直方图
        y1 = y - halfy;
        y2 = y + halfy;
        memset(histogram, 0, sizeof(int) * 256);

        for (i = y1, pcur = pgryimg + i * width; i <= y2; i++, pcur += width) {
            for (j = 0; j < halfx * 2 + 1; j++) {
                histogram[*(pcur + j)]++;
            }
        }

        // step.2-----初始化中值
        num = 0; // 记录着灰度值从0到中值的个数
        for (i = 0; i < 256; i++) {
            num += histogram[i];
            if (num * 2 > wsize) {
                med = i;
                break;
            }
        }

        // 滤波
        pres += halfx; // 没有处理图像左边界侧的像素
        for (x = halfx; x < width - halfx; x++) {
            // 赋值
            *(pres++) = med;

            // step.3-----直方图递推: 减去当前邻域最左边的一列,添加邻域右侧的一个新列
            j1 = x - halfx;     // 最左边列
            j2 = x + halfx + 1; // 右边的新列

            for (i = y1, pcur = pgryimg + i * width; i <= y2; i++, pcur += width) {
                // 减去最左边列
                v = *(pcur + j1);
                histogram[v]--;  // 更新直方图
                if (v <= med) num--; // 更新num

                // 添加右边的新列
                v = *(pcur + j2);
                histogram[v]++; // 更新直方图
                if (v <= med) num++; // 更新num
            }

            // step.4-----更新中值
            if (num * 2 < wsize) { // 到上次中值med的个数不够了,则med要变大
                for (med = med + 1; med < 256; med++) {
                    dbgcmptimes += 2; // 总的比较次数,调试用
                    num += histogram[med];
                    if (num * 2 > wsize) break;
                }
                dbgcmptimes += 1; // 总的比较次数,调试用
            }
            else { // 到上次中值med的个数多了,则med要变小
                while ((num - histogram[med]) * 2 > wsize) { // 若减去后,仍变小
                    dbgcmptimes++; // 总的比较次数,调试用
                    num -= histogram[med];
                    med--;
                }
                dbgcmptimes += 2; // 总的比较次数,调试用
            }
        }
        pres += halfx; // 没有处理图像右边界侧的像素
    }
    // 返回搜索中值需要的平均比较次数
    return dbgcmptimes * 1.0 / ((width - halfx * 2) * (height - halfy * 2));
}

//二值滤波
//pbinimg,  原始二值图像
// width, height,图像的宽度和高度
// m, n, 滤波邻域:m列n行
// threshold, 灰度阈值,大于等于该值时结果赋255
// presimg 结果图像
void rmwbinimgfilter(uint8_t* pbinimg,int width, int height,int m, int n,double threshold,uint8_t* presimg )
{
    // 没有对边界上邻域不完整的像素进行处理,可以采用变窗口的策略
    uint8_t* padd, * pdel, * pres;
    int halfx, halfy;
    int x, y, sum, sumthreshold;
    int sumcol[4096]; //约定图像宽度不大于4096

    // step.1------------初始化--------------------------//
    m = m / 2 * 2 + 1; //奇数化
    n = n / 2 * 2 + 1; //奇数化
    halfx = m / 2; //滤波器的x半径
    halfy = n / 2; //滤波器的y半径
    sumthreshold = max(1, (int)(threshold * m * n)); //转换成邻域内灰度值之和的阈值
    memset(sumcol, 0, sizeof(int) * width);
    for (y = 0, padd = pbinimg; y < n; y++)
    {
        for (x = 0; x < width; x++)
            sumcol[x] += *(padd++);
    }
    // step.2------------滤波----------------------------//
    for (y = halfy, pres = presimg + y * width, pdel = pbinimg; y < height - halfy; y++)
    {
        //初值
        for (sum = 0, x = 0; x < m; x++)
            sum += sumcol[x];
        //滤波
        pres += halfx; //跳过左侧
        for (x = halfx; x < width - halfx; x++)
        {
            //求灰度均值
            /*if (sum>=sumthreshold)
            {
                *(pres++) = 255;
            }
            else  *(pres++) = 0;*/
            *(pres++) = (sum >= sumthreshold) * 255; //请理解这个表达式的含义
            //换列,更新灰度和
            sum -= sumcol[x - halfx];     //减左边列
            sum += sumcol[x + halfx + 1]; //加右边列
        }
        pres += halfx; //跳过右侧
        //换行,更新sumcol
        for (x = 0; x < width; x++)
        {
            sumcol[x] -= *(pdel++); //减上一行
            sumcol[x] += *(padd++); //加下一行
        }
    }
    // step.3------------返回----------------------------//
    return;
}

//梯度算子加阈值
//pgryimg:输入的灰度图像数据指针。
//width:图像的宽度。
//height:图像的高度。
//pgrdimg:输出的梯度图像数据指针
void rmwgradientgryimg(uint8_t* pgryimg, int width, int height, uint8_t* pgrdimg)
{
    uint8_t* pgry, * pgrd;
    int dx, dy;
    int x, y;

    for (y = 0, pgry = pgryimg, pgrd = pgrdimg; y < height - 1; y++)
    {
        for (x = 0; x < width - 1; x++, pgry++)
        {
            dx = *pgry - *(pgry + 1);
            dy = *pgry - *(pgry + width);
            int gradient = (int)(sqrt(dx * dx * 1.0 + dy * dy));
            *pgrd++ = (gradient > 255) ? 255 : gradient;
        }
        *pgrd++ = 0; //尾列不做,边缘强度赋0
        pgry++;
    }
    memset(pgrd, 0, width); //尾行不做,边缘强度赋0
}

//梯度算子加阈值
//pgryimg:输入的灰度图像数据指针。
//width:图像的宽度。
//height:图像的高度。
//pgrdimg:输出的梯度图像数据指针
void rmwgradientgryimgplus(uint8_t* pgryimg, int width, int height, uint8_t* pgrdimg, int threshold)
{
    uint8_t* pgry, * pgrd;
    int dx, dy;
    int x, y;

    for (y = 0, pgry = pgryimg, pgrd = pgrdimg; y < height - 1; y++)
    {
        for (x = 0; x < width - 1; x++, pgry++)
        {
            dx = *pgry - *(pgry + 1);
            dy = *pgry - *(pgry + width);
            int gradient = (int)(sqrt(dx * dx * 1.0 + dy * dy));
            *(pgrd++) = (gradient > threshold) ? min(255, gradient) : 0;
        }
        *(pgrd++) = 0; //尾列不做,边缘强度赋0
        pgry++;
    }
    memset(pgrd, 0, width); //尾行不做,边缘强度赋0
}

//反相
void invertimage(uint8_t* image, int width, int height) {
    for (int i = 0; i < width * height; i++) {
        image[i] = 255 - image[i];
    }
}

//罗伯特算子
void rmwrobertsgryimg(uint8_t* pgryimg, int width, int height, uint8_t* prbtimg)
{
    uint8_t* pgry, * prbt;
    int dx, dy;
    int x, y;

    for (y = 0, pgry = pgryimg, prbt = prbtimg; y < height - 1; y++)
    {
        for (x = 0; x < width - 1; x++, pgry++)
        {
            dx = *pgry - *(pgry + width + 1);
            dy = *(pgry + 1) - *(pgry + width);
            *prbt++ = (uint8_t)(dx > dy ? dx : dy); // 使用三目运算符选择较大的值
        }
        *prbt++ = 0; // 尾列不做, 边缘强度赋0
        pgry++;
    }
    memset(prbt, 0, width); // 尾行不做, 边缘强度赋0
}

//索贝尔算子
void rmwsobelgryimg(uint8_t* pgryimg, int width, int height, uint8_t* psbimg)
{
    uint8_t* pgry, * psb;
    int dx, dy;
    int x, y;

    memset(psbimg, 0, width); // 首行不做, 边缘强度赋0
    for (y = 1, pgry = pgryimg + width, psb = psbimg + width; y < height - 1; y++)
    {
        *psb++ = 0; // 首列不做, 边缘强度赋0
        pgry++;
        for (x = 1; x < width - 1; x++, pgry++)
        {
            // 求dx
            dx = *(pgry - 1 - width) + (*(pgry - 1) * 2) + *(pgry - 1 + width);
            dx -= *(pgry + 1 - width) + (*(pgry + 1) * 2) + *(pgry + 1 + width);
            // 求dy
            dy = *(pgry - width - 1) + (*(pgry - width) * 2) + *(pgry - width + 1);
            dy -= *(pgry + width - 1) + (*(pgry + width) * 2) + *(pgry + width + 1);
            // 结果
            *psb++ = (uint8_t)min(255, abs(dx) + abs(dy));
        }
        *psb++ = 0; // 尾列不做, 边缘强度赋0
        pgry++;
    }
    memset(psb, 0, width); // 尾行不做, 边缘强度赋0
}

//prewitt算子
void rmwprewittgryimg(uint8_t* pgryimg, int width, int height, uint8_t* pprtimg)
{
    uint8_t* pgry, * pprt;
    int dx, dy, d45, d135, v1, v2;
    int x, y;

    memset(pprtimg, 0, width); // 首行不做, 边缘强度赋0
    for (y = 1, pgry = pgryimg + width, pprt = pprtimg + width; y < height - 1; y++)
    {
        *pprt++ = 0; // 首列不做, 边缘强度赋0
        pgry++;
        for (x = 1; x < width - 1; x++, pgry++)
        {
            // 求dx
            dx = *(pgry - 1 - width) + *(pgry - 1) + *(pgry - 1 + width);
            dx -= *(pgry + 1 - width) + *(pgry + 1) + *(pgry + 1 + width);
            // 求dy
            dy = *(pgry - width - 1) + *(pgry - width) + *(pgry - width + 1);
            dy -= *(pgry + width - 1) + *(pgry + width) + *(pgry + width + 1);
            // 求45度
            d45 = *(pgry - width - 1) + *(pgry - width) + *(pgry - 1);
            d45 -= *(pgry + width + 1) + *(pgry + width) + *(pgry + 1);
            // 求135度
            d135 = *(pgry - width) + *(pgry - width + 1) + *(pgry + 1);
            d135 -= *(pgry + width - 1) + *(pgry + width) + *(pgry - 1);
            // 结果
            v1 = abs(dx) > abs(dy) ? abs(dx) : abs(dy);
            v2 = abs(d45) > abs(d135) ? abs(d45) : abs(d135);
            *pprt++ = (uint8_t)((v1 > v2) ? ((v1 > 255) ? 255 : v1) : ((v2 > 255) ? 255 : v2));
        }
        *pprt++ = 0; // 尾列不做, 边缘强度赋0
        pgry++;
    }
    memset(pprt, 0, width); // 尾行不做, 边缘强度赋0
}

//沈俊算子
//pgryimg 和 ptmpimg 是指向 uint8_t 类型的指针,它们分别指向原始灰度图像数据和辅助图像数据。
//width 和 height 是整型参数,表示图像的宽度和高度。
//a0 是双精度浮点型参数,表示滤波系数。
//psjimg 是指向 uint8_t 类型的指针,它指向了输出的图像数据。
void rmwshenjungryimg(uint8_t* pgryimg,uint8_t* ptmpimg, int width, int height, double a0, uint8_t* psjimg)
{
    uint8_t* pgry, * pcur, * psj, * pend;
    int lut[512], * alut; // a0查找表
    int x, y, pre, dif;

    // step 1: 初始化查找表
    a0 = (a0 < 0.01) ? 0.01 : ((a0 > 0.99) ? 0.99 : a0); // 安全性检查
    // a0查找表, 进行了四舍五入
    alut = lut + 256;
    for (alut[0] = 0, dif = 1; dif < 256; dif++)
    {
        alut[dif] = (int)(dif * a0 + 0.5);
        alut[-dif] = (int)(-dif * a0 - 0.5);
    }

    // step 2: 递推实现指数滤波
    // 按行滤波
    for (y = 0, pgry = pgryimg, pcur = ptmpimg; y < height; y++)
    {
        // 从左向右: p1(y,x) = p1(y,x-1) + a * [p(y,x) - p1(y,x-1)]
        *(pcur++) = pre = *(pgry++);
        for (x = 1; x < width; x++, pgry++)
            *(pcur++) = pre = pre + alut[*pgry - pre];
        pcur--; // 回到行尾
        // 从右向左: p2(y,x) = p2(y,x+1) - a * [p1(y,x) - p2(y,x+1)]
        for (x = width - 2, pcur = pcur - 1; x >= 0; x--)
            *(pcur--) = pre = pre + alut[*pcur - pre];
        pcur += (width + 1); // 回到下一行的开始
    }
    // 按列滤波
    for (x = 0, pcur = ptmpimg; x < width; x++, pcur = ptmpimg + x)
    {
        // 从上向下: p3(y,x) = p3(y-1,x) + a * [p2(y,x) - p3(y-1,x)]
        pre = *pcur;
        for (y = 1, pcur += width; y < height; y++, pcur += width)
            *pcur = pre = pre + alut[*pcur - pre];
        pcur -= width; // 回到列尾
        // 从下向上: p4(i,j) = p4(i+1,j) + a * [p3(i,j) - p4(i+1,j)]
        for (y = height - 2, pcur -= width; y >= 0; y--, pcur -= width)
            *pcur = pre = pre + alut[*pcur - pre];
    }

    // step 3: 正导数=1,负导数为0,0必须也是0
    pend = ptmpimg + width * height;
    for (pcur = ptmpimg, pgry = pgryimg; pcur < pend; pgry++)
    {
        *(pcur++) = (*pcur > *pgry);
    }

    // step 4: 过零点检测
    memset(psjimg, 0, width * height); // 边缘强度赋0
    psj = psjimg + width;
    pcur = ptmpimg + width; // 首行不做 
    for (y = 1; y < height - 1; y++)
    {
        psj++; pcur++;  // 首列不做
        for (x = 1; x < width - 1; x++, pgry++, pcur++, psj++)
        {
            if (*pcur) // 正导数
            {
                // 下面使用4邻域, 边缘为8连通, 不能保证4连通; 使用8邻域才能保证边缘4连通
                if ((!*(pcur - 1)) || // 左, 必须<=0, 不能<0
                    (!*(pcur + 1)) || // 右, 必须<=0, 不能<0
                    (!*(pcur - width)) || // 上, 必须<=0, 不能<0
                    (!*(pcur + width)))   // 下, 必须<=0, 不能<0
                {
                    *psj = 255; // 周围有导数小于等于0
                }
            }
        }
        psj++; pcur++;  // 尾列不做
    }
}

//沈俊算子加索贝尔算子
//pgryimg:指向原始灰度图像数据的指针
//ptmpimg:指向辅助图像数据的指针
//width:图像的宽度
//height:图像的高度
//a0:这是沈俊算子的参数,用于控制边缘检测的灵敏度。
//grdthre:这是sobel算子的梯度阈值
//pedgeimg:最终边缘图像数据的指针
void rmwextractriceedge(uint8_t* pgryimg,uint8_t* ptmpimg,int width,int height,double a0, int grdthre, uint8_t* pedgeimg)
{
    // step.1------------沈俊算子-----------------------//
    rmwshenjungryimg(pgryimg, ptmpimg, width, height, a0, pedgeimg);
    // step.2------------sobel算子----------------------//
    rmwsobelgryimg(pgryimg, width, height, ptmpimg);
    // step.3------------二者融合-----------------------//
    for (int i = 0; i < width * height; i++)
    {
        *(pedgeimg + i) = (pedgeimg[i] && (ptmpimg[i] > grdthre)) * 255;
    }
    // step.4------------结束---------------------------//
    return;
}
(0)
打赏 微信扫一扫 微信扫一扫

您想发表意见!!点此发布评论

推荐阅读

Stable Diffusion1.5网络结构-超详细原创

08-01

内容产品运营方案业务架构解析与实践探索

08-01

应聘产品运营的个人简历(精选篇)

08-01

AI绘画ComfyUI 进阶教程 | 字节最强换脸插件PuLID 详解,还请收藏!

07-31

Stable Diffusion - 从零到一学习AI绘画,电脑配置和安装。

07-31

还没用过这5个AI绘画工具吗 ?特别好用

07-31

猜你喜欢

版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。

发表评论