Java OpenCV 图像白平衡算法

原文链接:https://blog.csdn.net/dcrmg/article/details/53545510
        彩色相机内部有三个CCD电子耦合元件,分别用来感受红绿蓝三中颜色的光线,默认情况下,三个颜色的感光电路信号的放大比例是1:1:1的。在理想的拍摄环境下,纯白色的RGB分量按照1:1:1的比例放大之后,得到的是没有偏色的白色,当在非理想的环境光条件下,白色成像出来的效果会偏向环境光的颜色,而不是纯白色。

        相机的自动白平衡就是要通过改变RGB感光电路信号的放大比例,让受环境光影响的白色还原成纯白色,保证在各种光线条件下,成像色彩跟物体真实的色彩保持一致。

        常用的一种自动白平衡算法是灰度世界法,它假设对于一副色彩丰富的图像,图像上RGB三个分量的平均值趋于同一个灰度值,一般取这个灰度值的大小为RGB三分量的平均值。

代码

public static void main(String[] args) {
    System.load("D:\\opencv-320\\opencv_java320.dll");

    JavaTest javaTest = new JavaTest();

    Mat src = Imgcodecs.imread("F:\\opencvPhoto\\oneWord\\test2.jpg");
    List<Mat> splitMat = new ArrayList<Mat>();
    Core.split(src, splitMat);
    double meanB = Core.mean(splitMat.get(0)).val[0];
    double meanG = Core.mean(splitMat.get(1)).val[0];
    double meanR = Core.mean(splitMat.get(2)).val[0];

    double kB = (meanB + meanG + meanR) / (3 * meanB);
    double kG = (meanB + meanG + meanR) / (3 * meanG);
    double kR = (meanB + meanG + meanR) / (3 * meanR);

    List<Mat> mergeMatList = new ArrayList<Mat>();
    Mat merge = new Mat();
    mergeMatList.add(javaTest.changeRGB(splitMat.get(0), kB));
    mergeMatList.add(javaTest.changeRGB(splitMat.get(1), kG));
    mergeMatList.add(javaTest.changeRGB(splitMat.get(2), kR));

    Core.merge(mergeMatList, merge);

    Imgcodecs.imwrite("F:\\opencvPhoto\\oneWord\\xx.jpg", merge); 
}

public Mat changeRGB(Mat mat, double k) {
    Mat newMat = new Mat(mat.size(), CvType.CV_8UC1);
    for (int i = 0; i < mat.rows(); i++) {
        for (int j = 0; j < mat.cols(); j++) {
            double[] color = mat.get(i, j);
            color[0] *= k;
            newMat.put(i, j, color);
        }
    }
    return newMat;
}

原图1
在这里插入图片描述效果1
在这里插入图片描述
图片2
在这里插入图片描述
效果2
在这里插入图片描述

发布了90 篇原创文章 · 获赞 9 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/wangwenjie1997/article/details/104594309