#include<iostream>

#include <opencv2/opencv.hpp>

#include <math.h>

using namespace cv;

using namespace std;

////该代码实现白平衡算法中的灰度世界法，能有效改善图像发红发蓝发绿的现象；

#include <opencv2/opencv.hpp>

using namespace cv;

int main()

{

    Mat g_srcImage,dstImage;

    vector<Mat> g_vChannels;

    g_srcImage = imread("/home/leoxae/KeekoRobot/TestPic/multCards/58.png");

    imshow("原图",g_srcImage);

    //waitKey(0);

    //分离通道

    split(g_srcImage,g_vChannels);

    Mat imageBlueChannel = g_vChannels.at(0);

    Mat imageGreenChannel = g_vChannels.at(1);

    Mat imageRedChannel = g_vChannels.at(2);

    double imageBlueChannelAvg=0;

    double imageGreenChannelAvg=0;

    double imageRedChannelAvg=0;

    //求各通道的平均值

    imageBlueChannelAvg = mean(imageBlueChannel)[0];

    imageGreenChannelAvg = mean(imageGreenChannel)[0];

    imageRedChannelAvg = mean(imageRedChannel)[0];

    //求出个通道所占增益

    double K = (imageRedChannelAvg+imageGreenChannelAvg+imageRedChannelAvg)/3;

    double Kb = K/imageBlueChannelAvg;

    double Kg = K/imageGreenChannelAvg;

    double Kr = K/imageRedChannelAvg;

    //更新白平衡后的各通道BGR值

    addWeighted(imageBlueChannel,Kb,0,0,0,imageBlueChannel);

    addWeighted(imageGreenChannel,Kg,0,0,0,imageGreenChannel);

    addWeighted(imageRedChannel,Kr,0,0,0,imageRedChannel);

    merge(g_vChannels,dstImage);//图像各通道合并

    imshow("白平衡后图",dstImage);

    imwrite("/home/leoxae/KeekoRobot/TestPic/multCards/test.png",dstImage);

    waitKey(0);

    return 0;

}