MultiScaleDetailBoosting.cpp

Author: BBuf

MultiScaleDetailBoosting.cpp

@@ -0,0 +1,112 @@
+#include "opencv2/opencv.hpp"
+#include "iostream"
+#include "algorithm"
+#include "vector"
+#include "stdio.h"
+using namespace std;
+using namespace cv;
+
+void separateGaussianFilter(const Mat &src, Mat &dst, int ksize, double sigma){
+    CV_Assert(src.channels()==1 || src.channels() == 3); //只处理单通道或者三通道图像
+    //生成一维的
+    double *matrix = new double[ksize];
+    double sum = 0;
+    int origin = ksize / 2;
+    for(int i = 0; i < ksize; i++){
+        double g = exp(-(i-origin) * (i-origin) / (2 * sigma * sigma));
+        sum += g;
+        matrix[i] = g;
+    }
+    for(int i = 0; i < ksize; i++) matrix[i] /= sum;
+    int border = ksize / 2;
+    copyMakeBorder(src, dst, border, border, border, border, BORDER_CONSTANT);
+    int channels = dst.channels();
+    int rows = dst.rows - border;
+    int cols = dst.cols - border;
+    //水平方向
+    for(int i = border; i < rows; i++){
+        for(int j = border; j < cols; j++){
+            double sum[3] = {0};
+            for(int k = -border; k<=border; k++){
+                if(channels == 1){
+                    sum[0] += matrix[border + k] * dst.at<uchar>(i, j+k);
+                }else if(channels == 3){
+                    Vec3b rgb = dst.at<Vec3b>(i, j+k);
+                    sum[0] += matrix[border+k] * rgb[0];
+                    sum[1] += matrix[border+k] * rgb[1];
+                    sum[2] += matrix[border+k] * rgb[2];
+                }
+            }
+            for(int k = 0; k < channels; k++){
+                if(sum[k] < 0) sum[k] = 0;
+                else if(sum[k] > 255) sum[k] = 255;
+            }
+            if(channels == 1)
+                dst.at<Vec3b>(i, j) = static_cast<uchar>(sum[0]);
+            else if(channels == 3){
+                Vec3b rgb = {static_cast<uchar>(sum[0]), static_cast<uchar>(sum[1]), static_cast<uchar>(sum[2])};
+                dst.at<Vec3b>(i, j) = rgb;
+            }
+        }
+    }
+    //竖直方向
+    for(int i = border; i < rows; i++){
+        for(int j = border; j < cols; j++){
+            double sum[3] = {0};
+            for(int k = -border; k<=border; k++){
+                if(channels == 1){
+                    sum[0] += matrix[border + k] * dst.at<uchar>(i+k, j);
+                }else if(channels == 3){
+                    Vec3b rgb = dst.at<Vec3b>(i+k, j);
+                    sum[0] += matrix[border+k] * rgb[0];
+                    sum[1] += matrix[border+k] * rgb[1];
+                    sum[2] += matrix[border+k] * rgb[2];
+                }
+            }
+            for(int k = 0; k < channels; k++){
+                if(sum[k] < 0) sum[k] = 0;
+                else if(sum[k] > 255) sum[k] = 255;
+            }
+            if(channels == 1)
+                dst.at<Vec3b>(i, j) = static_cast<uchar>(sum[0]);
+            else if(channels == 3){
+                Vec3b rgb = {static_cast<uchar>(sum[0]), static_cast<uchar>(sum[1]), static_cast<uchar>(sum[2])};
+                dst.at<Vec3b>(i, j) = rgb;
+            }
+        }
+    }
+    delete [] matrix;
+}
+
+Mat MultiScaleDetailBoosting(Mat src, int Radius){
+    int rows = src.rows;
+    int cols = src.cols;
+    Mat B1, B2, B3;
+    separateGaussianFilter(src, B1, Radius, 1.0);
+    separateGaussianFilter(src, B2, Radius*2-1, 2.0);
+    separateGaussianFilter(src, B3, Radius*4-1, 4.0);
+    float w1 = 0.5, w2 = 0.5, w3 = 0.25;
+    Mat dst(rows, cols, CV_8UC3);
+    for(int i = 0; i < rows; i++){
+        for(int j = 0; j < cols; j++){
+            for(int k = 0; k < 3; k++){
+                int D1 = src.at<Vec3b>(i, j)[k] - B1.at<Vec3b>(i, j)[k];
+                int D2 = B1.at<Vec3b>(i, j)[k] - B2.at<Vec3b>(i, j)[k];
+                int D3 = B2.at<Vec3b>(i, j)[k] - B3.at<Vec3b>(i, j)[k];
+                int sign = D1 > 0 ? 1 : -1;
+                dst.at<Vec3b>(i, j)[k] = saturate_cast<uchar>((1 - w1*sign) * D1 - w2 * D2 + w3 * D3 + src.at<Vec3b>(i, j)[k]);
+            }
+        }
+    }
+    return dst;
+}
+
+int main(){
+    Mat src = imread("../tmp.png");
+    Mat dst = MultiScaleDetailBoosting(src, 5);
+    imshow("origin", src);
+    imshow("result", dst);
+    waitKey(0);
+    imwrite("../result.jpg", dst);
+    return 0;
+}

README.md

@@ -10,6 +10,7 @@
 - Optimized contrast enhancement for real-time image and video dehazin.cpp C++复现了《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
+- MultiScaleDetailBoosting.cpp C++复现了《DARK IMAGE ENHANCEMENT BASED ON PAIRWISE TARGET CONTRAST AND MULTI-SCALE DETAIL BOOSTING》论文，可以用于提升图像不同程度的细节信息。算法原理请看：https://blog.csdn.net/just_sort/article/details/85007555
 
 # ImageFiletering 复现了一些图像滤波算法