Add Real-time adaptive contrast enhancement for imaging sensors.cpp

Author: BBuf

README.md

@@ -26,3 +26,4 @@
 - Inrbl.cpp C++复现了《改进非线性亮度提升模型的逆光图像恢复》这篇论文，可以做逆光图像恢复。算法原理请看：https://blog.csdn.net/just_sort/article/details/86681325
 - unevenLightCompensate.cpp C++复现了《一种基于亮度均衡的图像阈值分割技术》这篇论文的光照补偿部分，可以对光照不均匀，曝光，逆光图像做亮度均衡，效果不错。原理请看：https://blog.csdn.net/just_sort/article/details/88551771
 - Adaptive correction algorithm for illumination inhomogeneity image based on two-dimensional gamma function.cpp C++复现了《基于二维伽马函数的光照不均匀图像自适应校正算法》这篇论文，对光照不均匀的图像有较好的校正效果，且不会像Retiex那样出现光晕。原理请看：https://blog.csdn.net/just_sort/article/details/88569129
+- Real-time adaptive contrast enhancement for imaging sensors.cpp C++复现了《Real-time adaptive contrast enhancement for imaging sensors》这篇论文，实时自适应局部对比度增强算法。原理请看：https://blog.csdn.net/just_sort/article/details/85208124

Real-time adaptive contrast enhancement for imaging sensors.cpp

@@ -0,0 +1,56 @@
+//自适应对比度增强算法，C表示对高频的直接增益系数,n表示滤波半径，maxCG表示对CG做最大值限制
+Mat ACE(Mat src, int C = 3, int n = 3, float MaxCG = 7.5){
+    int row = src.rows;
+    int col = src.cols;
+    Mat meanLocal; //图像局部均值
+    Mat varLocal; //图像局部方差
+    Mat meanGlobal; //全局均值
+    Mat varGlobal; //全局标准差
+    blur(src.clone(), meanLocal, Size(n, n));
+    Mat highFreq = src - meanLocal;
+    varLocal = highFreq.mul(highFreq);
+    varLocal.convertTo(varLocal, CV_32F);
+    for(int i = 0; i < row; i++){
+        for(int j = 0; j < col; j++){
+            varLocal.at<float>(i, j) = (float)sqrt(varLocal.at<float>(i, j));
+        }
+    }
+    meanStdDev(src, meanGlobal, varGlobal);
+    Mat gainArr = meanGlobal / varLocal; //增益系数矩阵
+    for(int i = 0; i < row; i++){
+        for(int j = 0; j < col; j++){
+            if(gainArr.at<float>(i, j) > MaxCG){
+                gainArr.at<float>(i, j) = MaxCG;
+            }
+        }
+    }
+    printf("%d %d\n", row, col);
+    gainArr.convertTo(gainArr, CV_8U);
+    gainArr = gainArr.mul(highFreq);
+    Mat dst1 = meanLocal + gainArr;
+    Mat dst2 = meanLocal + C * highFreq;
+    return dst1;
+}
+
+int main(){
+    Mat src = imread("../test.png");
+    vector <Mat> now;
+    split(src, now);
+    int C = 150;
+    int n = 5;
+    float MaxCG = 3;
+    Mat dst1 = ACE(now[0], C, n, MaxCG);
+    Mat dst2 = ACE(now[1], C, n, MaxCG);
+    Mat dst3 = ACE(now[2], C, n, MaxCG);
+    now.clear();
+    Mat dst;
+    now.push_back(dst1);
+    now.push_back(dst2);
+    now.push_back(dst3);
+    cv::merge(now, dst);
+    imshow("origin", src);
+    imshow("result", dst);
+    imwrite("../result.jpg", dst);
+    waitKey(0);
+    return 0;
+}