前置内容

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

灰度世界算法原理

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

算法步骤

确定Gray有2种方法，一种是取固定值，比如最亮灰度值的一般，8位显示为128。另一种就是通过计算图像R,G,B的三个通道 $<mover accent="true"> R ˉ </mover>$ , $<mover accent="true"> G ˉ </mover>$ , $<mover accent="true"> B ˉ </mover>$ ，取 $<mover accent="true"> G ˉ </mover> r a y = \frac{<mover accent="true"> R ˉ </mover> + <mover accent="true"> G ˉ </mover> + <mover accent="true"> B ˉ </mover>}{3}$
计算 $R$ , $G$ , $B$ ，3个通道的增益系数： $k_{r} = \frac{<mover accent="true"> G ˉ </mover> r a y}{<mover accent="true">}$ , $k_{g} = \frac{<mover accent="true"> G ˉ </mover> r a y}{<mover accent="true">}$ , $k_{b} = \frac{<mover accent="true"> G ˉ </mover> r a y}{<mover accent="true">}$
根据Von Kries对角模型，对于图像中的每个像素C，调整其分量R,G,B分量： $C (R^{'}) = C (R) * k_{r}$ , $C (G^{'}) = C (G) * k_{g}$ , $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;
}

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

前置内容

灰度世界算法原理

算法步骤

算法优缺点

源码实现

效果