Add Contrast Image Correction Method.cpp

Author: BBuf

Contrast Image Correction Method.cpp

@@ -0,0 +1,80 @@
+#include "opencv2/opencv.hpp"
+#include "iostream"
+#include "algorithm"
+#include "vector"
+#include "stdio.h"
+using namespace std;
+using namespace cv;
+
+Mat ContrastImageCorrection(Mat src){
+    int rows = src.rows;
+    int cols = src.cols;
+    Mat yuvImg;
+    cvtColor(src, yuvImg, CV_BGR2YUV_I420);
+    vector <Mat> mv;
+    split(yuvImg, mv);
+    Mat OldY = mv[0].clone();
+//    for(int i = 0; i < rows; i++){
+//        for(int j = 0; j < cols; j++){
+//            mv[0].at<uchar>(i, j) = 255 - mv[0].at<uchar>(i, j);
+//        }
+//    }
+    Mat temp;
+    bilateralFilter(mv[0], temp, 9, 50, 50);
+    //GaussianBlur(mv[0], temp, Size(41, 41), BORDER_DEFAULT);
+    for(int i = 0; i < rows; i++){
+        for(int j = 0; j < cols; j++){
+            float Exp = pow(2, (128 - (255 - temp.at<uchar>(i, j))) / 128.0);
+            int value = int(255 * pow(OldY.at<uchar>(i, j) / 255.0, Exp));
+            temp.at<uchar>(i, j) = value;
+        }
+    }
+    Mat dst(rows, cols, CV_8UC3);
+//    mv[0] = temp;
+//    merge(mv, dst);
+//    cvtColor(dst, dst, CV_YUV2BGRA_I420);
+    for(int i = 0; i < rows; i++){
+        for(int j = 0; j < cols; j++) {
+            if (OldY.at<uchar>(i, j) == 0) {
+                for (int k = 0; k < 3; k++) dst.at<Vec3b>(i, j)[k] = 0;
+            } else {
+                //channel B
+                dst.at<Vec3b>(i, j)[0] =
+                        (temp.at<uchar>(i, j)  * (src.at<Vec3b>(i, j)[0] + OldY.at<uchar>(i, j)) / OldY.at<uchar>(i, j) +
+                         src.at<Vec3b>(i, j)[0] - OldY.at<uchar>(i, j)) >> 1;
+                //channel G
+                dst.at<Vec3b>(i, j)[1] =
+                        (temp.at<uchar>(i, j)  * (src.at<Vec3b>(i, j)[1] + OldY.at<uchar>(i, j)) / OldY.at<uchar>(i, j) +
+                         src.at<Vec3b>(i, j)[1] - OldY.at<uchar>(i, j)) >> 1;
+                //channel R
+                dst.at<Vec3b>(i, j)[2] =
+                        (temp.at<uchar>(i, j) * (src.at<Vec3b>(i, j)[2] + OldY.at<uchar>(i, j)) / OldY.at<uchar>(i, j) +
+                         src.at<Vec3b>(i, j)[2] - OldY.at<uchar>(i, j)) >> 1;
+            }
+        }
+    }
+//    for(int i = 0; i < rows; i++){
+//        for(int j = 0; j < cols; j++){
+//            for(int k = 0; k < 3; k++){
+//                if(dst.at<Vec3b>(i, j)[k] < 0){
+//                    dst.at<Vec3b>(i, j)[k] = 0;
+//                }else if(dst.at<Vec3b>(i, j)[k] > 255){
+//                    dst.at<Vec3b>(i, j)[k] = 255;
+//                }
+//            }
+//        }
+//    }
+    return dst;
+}
+
+int main(){
+    Mat src = imread("/home/streamax/CLionProjects/Paper/1.jpg");
+    Rect rect(0, 0, (src.cols-1)/2*2, (src.rows-1)/2*2);
+    Mat newsrc = src(rect);
+    Mat dst = ContrastImageCorrection(newsrc);
+    imshow("origin", newsrc);
+    imshow("result", dst);
+    cv::imwrite("../result.jpg", dst);
+    waitKey(0);
+    return 0;
+}

README.md

@@ -1,14 +1,15 @@
 # 记录一些图像处理中的论文及代码复现
 
-- Retinex MSRCR.cpp 带色彩恢复的多尺度视网膜增强算法 学习链接：http://www.cnblogs.com/Imageshop/archive/2013/04/17/3026881.html
-- ImageDataIncrease.cpp 常见的图片数据扩充，包括一些PS算法 具体为旋转，添加高斯，椒盐噪声，增加老照片效果，增加和降低图像饱和度，对原图缩放，亮度增强，对比度增强，磨皮美白，偏色矫正，同态滤波，过曝，灰度化，轮换通道，图像错切，运动模糊，钝化蒙版，PS滤镜算法之球面化 (凸出和凹陷效果)
-- HDR.cpp 对《Adaptive Local Tone Mapping Based on Retinex for High Dynamic Range Images》C++朴素实现，无优化，详情请查看https://blog.csdn.net/just_sort/article/details/84030723
-- Adaptive Logarithmic Mapping For Displaying High Contrast Scenes.cpp 《Adaptive Logarithmic Mapping For Displaying High Contrast Scenes》c++朴素实现 ,详情请看：https://blog.csdn.net/just_sort/article/details/84066390
-- Single Image Haze Removal Using Dark Channel Prior.cpp 《Single Image Haze Removal Using Dark Channel Prior》论文阅读及复现，详情请看：https://blog.csdn.net/just_sort/article/details/84110518
-- Local Color Correction.cpp C++复现了《Local Color Correction》论文，详情请参考：https://blog.csdn.net/just_sort/article/details/84539295
-- PartialcolorJudge.cpp C++复现了《基于图像分析的偏色检测及颜色校正方法》论文，实现快速判断图片是否存在偏色，详情请看：https://blog.csdn.net/just_sort/article/details/84897976
-- Optimized contrast enhancement for real-time image and video dehazin.cpp 复现了《Optimized contrast enhancement for real-time image and video dehazin》这篇论文，相对于He Kaiming的暗通道去雾，对天空具有天然的免疫力。详情请看：https://blog.csdn.net/just_sort/article/details/84932848
-- AutoLevelAndAutoContrast.cpp C++复现了自动色阶调整和自动对比度调整，其中自动色阶调整可以用于去雾和水下图像恢复，算法原理请看：https://www.cnblogs.com/Imageshop/archive/2011/11/13/2247614.html
+- Retinex MSRCR.cpp 带色彩恢复的多尺度视网膜增强算法。 算法原理请看：http://www.cnblogs.com/Imageshop/archive/2013/04/17/3026881.html
+- ImageDataIncrease.cpp 常见的图片数据扩充。包括一些PS算法 具体为旋转，添加高斯，椒盐噪声，增加老照片效果，增加和降低图像饱和度，对原图缩放，亮度增强，对比度增强，磨皮美白，偏色矫正，同态滤波，过曝，灰度化，轮换通道，图像错切，运动模糊，钝化蒙版，PS滤镜算法之球面化 (凸出和凹陷效果)
+- HDR.cpp 对《Adaptive Local Tone Mapping Based on Retinex for High Dynamic Range Images》C++朴素实现，无优化。算法原理请看：https://blog.csdn.net/just_sort/article/details/84030723
+- Adaptive Logarithmic Mapping For Displaying High Contrast Scenes.cpp 《Adaptive Logarithmic Mapping For Displaying High Contrast Scenes》c++朴素实现 。算法原理请看：https://blog.csdn.net/just_sort/article/details/84066390
+- Single Image Haze Removal Using Dark Channel Prior.cpp 《Single Image Haze Removal Using Dark Channel Prior》论文阅读及复现。算法原理请看：https://blog.csdn.net/just_sort/article/details/84110518
+- Local Color Correction.cpp C++复现了《Local Color Correction》论文。算法原理请看：https://blog.csdn.net/just_sort/article/details/84539295
+- PartialcolorJudge.cpp C++复现了《基于图像分析的偏色检测及颜色校正方法》论文，实现快速判断图片是否存在偏色。算法原理请看：https://blog.csdn.net/just_sort/article/details/84897976
+- Optimized contrast enhancement for real-time image and video dehazin.cpp 复现了《Optimized contrast enhancement for real-time image and video dehazin》这篇论文，相对于He Kaiming的暗通道去雾，对天空具有天然的免疫力。算法原理请看：https://blog.csdn.net/just_sort/article/details/84932848
+- AutoLevelAndAutoContrast.cpp C++复现了自动色阶调整和自动对比度调整，其中自动色阶调整可以用于去雾和水下图像恢复。算法原理请看：https://www.cnblogs.com/Imageshop/archive/2011/11/13/2247614.html
+- Contrast Image Correction Method.cpp C++复现了《Contrast Image Correction Method》这篇论文，可以自适应矫正图像。算法原理请看：https://blog.csdn.net/just_sort/article/details/85005510
 
 # ImageFiletering 复现了一些图像滤波算法