opencv 彩色图片自动白平衡 c++代码

下面是使用 OpenCV 库实现彩色图像自动白平衡的 C++ 代码示例：

#include <opencv2/opencv.hpp>

using namespace cv;

void autoWhiteBalance(Mat& src, Mat& dst) {
    // 将图像转换为 LAB 色彩空间
    Mat labImage;
    cvtColor(src, labImage, COLOR_BGR2Lab);

    // 计算 a 通道和 b 通道的平均值
    Scalar avgPixel = mean(labImage);
    float avgA = avgPixel[1];
    float avgB = avgPixel[2];

    // 计算需要调整的 a 通道和 b 通道的增益
    float ka = 128 / avgA;
    float kb = 128 / avgB;

    // 应用增益调整
    std::vector<Mat> channels;
    split(labImage, channels);
    channels[1] = channels[1] * ka;
    channels[2] = channels[2] * kb;
    merge(channels, dst);

    // 将图像转换回 BGR 色彩空间
    cvtColor(dst, dst, COLOR_Lab2BGR);
}

int main() {
    Mat src = imread("input.jpg");
    if (src.empty()) {
        return -1;
    }

    Mat dst;
    autoWhiteBalance(src, dst);

    imshow("input", src);
    imshow("output", dst);
    waitKey(0);

    return 0;
}

该代码使用 `autoWhiteBalance` 函数对输入图像进行自动白平衡处理，其中：

1. 将输入图像转换为 LAB 色彩空间。
2. 计算 a 通道和 b 通道的平均值。
3. 根据平均值计算需要调整的 a 通道和 b 通道的增益。
4. 将增益应用到 a 通道和 b 通道上，得到平衡后的图像。
5. 将平衡后的图像转换回 BGR 色彩空间。

在 `main` 函数中，读取输入图像并调用 `autoWhiteBalance` 函数进行处理，然后展示原始图像和平衡后的图像。