灰度世界算法原理及C++实现

版权声明:蒟蒻博客,请随意转载,但请保留原地址。 https://blog.csdn.net/just_sort/article/details/85638420

前置内容

人的视觉系统具有颜色恒常性,能从变化的光照环境和成像条件下获取物体表面颜色的不变特性,但成像设备并不具有这样的调节功能,不同的光照环境会导致采集到的图像颜色与真实颜色存在一定程度的偏差,需要选择合适的颜色平衡算法去消除光照环境对颜色显示的影响。

灰度世界算法原理

灰度世界算法以灰度世界假设为基础,假设为:对于一幅有着大量色彩变化的图像,R,G,B,3个分量的平均值趋于同一个灰度值 G ˉ r a y \bar Gray 。从物理意思上讲,灰度世界算法假设自然界景物对于光线的平均反射的均值在整体上是一个定值,这个定值近似为“灰色”。颜色平衡算法将这一假设强制应用于待处理的图像,可以从图像中消除环境光的影响,获得原始场景图像。

算法步骤

  • 确定Gray有2种方法,一种是取固定值,比如最亮灰度值的一般,8位显示为128。另一种就是通过计算图像R,G,B的三个通道 R ˉ \bar R , G ˉ \bar G , B ˉ \bar B ,取 G ˉ r a y = R ˉ + G ˉ + B ˉ 3 \bar Gray=\frac{\bar R + \bar G + \bar B}{3}
  • 计算 R R , G G , B B ,3个通道的增益系数: k r = G ˉ r a y R ˉ k_r=\frac{\bar Gray}{\bar R} , k g = G ˉ r a y G ˉ k_g=\frac{\bar Gray}{\bar G} , k b = G ˉ r a y B ˉ k_b=\frac{\bar Gray}{\bar B}
  • 根据Von Kries对角模型,对于图像中的每个像素C,调整其分量R,G,B分量: C ( R ) = C ( R ) k r C(R')=C(R)*k_r , C ( G ) = C ( G ) k g C(G')=C(G)*k_g , C ( B ) = C ( B ) k b C'(B)=C(B)*k_b

算法优缺点

此算法简单快速,但是当图像场景颜色并不丰富时,尤其诗出现大量单色物体时,该算法会失效。

源码实现

Mat GrayWorld(const Mat &src){
    vector <Mat> bgr;
    cv::split(src, bgr);
    double B = 0;
    double G = 0;
    double R = 0;
    int row = src.rows;
    int col = src.cols;
    Mat dst(row, col, CV_8UC3);
    for(int i = 0; i < row; i++){
        for(int j = 0; j < col; j++){
              B += 1.0 * src.at<Vec3b>(i, j)[0];
              G += 1.0 * src.at<Vec3b>(i, j)[1];
              R += 1.0 * src.at<Vec3b>(i, j)[2];
        }
    }
    B /= (row * col);
    G /= (row * col);
    R /= (row * col);
    printf("%.5f %.5f %.5f\n", B, G, R);
    double GrayValue = (B + G + R) / 3;
    printf("%.5f\n", GrayValue);
    double kr = GrayValue / R;
    double kg = GrayValue / G;
    double kb = GrayValue / B;
    printf("%.5f %.5f %.5f\n", kb, kg, kr);
    for(int i = 0; i < row; i++){
        for(int j = 0; j < col; j++){
            dst.at<Vec3b>(i, j)[0] = (int)(kb * src.at<Vec3b>(i, j)[0]);
            dst.at<Vec3b>(i, j)[1] = (int)(kg * src.at<Vec3b>(i, j)[1]);
            dst.at<Vec3b>(i, j)[2] = (int)(kr * src.at<Vec3b>(i, j)[2]);
            for(int k = 0; k < 3; k++){
                if(dst.at<Vec3b>(i, j)[k] > 255){
                    dst.at<Vec3b>(i, j)[k] = 255;
                }
            }
        }
    }
    return dst;
}

效果

在这里插入图片描述在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/just_sort/article/details/85638420