opencv 彩色图Lens Shading Correction c++代码

以下是使用OpenCV进行彩色图像Lens Shading校正的C++代码示例：

#include <opencv2/opencv.hpp>

using namespace cv;

void lensShadingCorrection(Mat& src, Mat& dst, float alpha)
{
    CV_Assert(src.channels() == 3);

    Mat_<Vec3f> src_f, dst_f;
    src.convertTo(src_f, CV_32FC3, 1.0 / 255.0);
    dst.create(src.size(), CV_32FC3);
    dst_f.create(src.size(), CV_32FC3);

    // 计算每个通道的均值
    Scalar B, G, R;
    B = mean(src_f(0));
    G = mean(src_f(1));
    R = mean(src_f(2));

    // 计算每个像素的校正系数
    Mat_<float> L(src.size(), 1);
    for (int r = 0; r < src.rows; r++)
    {
        float* L_data = L.ptr<float>(r);
        for (int c = 0; c < src.cols; c++)
        {
            float b = src_f(r, c)[0];
            float g = src_f(r, c)[1];
            float r = src_f(r, c)[2];

            float l = sqrtf((B[0] / b) * (G[0] / g) * (R[0] / r));
            L_data[c] = l;
        }
    }

    // 应用校正
    for (int r = 0; r < src.rows; r++)
    {
        Vec3f* src_data = src_f.ptr<Vec3f>(r);
        Vec3f* dst_data = dst_f.ptr<Vec3f>(r);
        float* L_data = L.ptr<float>(r);
        for (int c = 0; c < src.cols; c++)
        {
            float l = L_data[c];
            dst_data[c][0] = powf(src_data[c][0] * l, alpha);
            dst_data[c][1] = powf(src_data[c][1] * l, alpha);
            dst_data[c][2] = powf(src_data[c][2] * l, alpha);
        }
    }

    dst_f.convertTo(dst, CV_8UC3, 255.0);
}

int main()
{
    Mat image = imread("input.jpg");

    Mat corrected;
    lensShadingCorrection(image, corrected, 0.8f);

    imshow("Input", image);
    imshow("Corrected", corrected);
    waitKey();

    return 0;
}

该函数将输入图像src进行校正，并将结果存储在输出图像dst中。参数alpha控制校正程度，通常介于0.5到1.0之间。