Add PS色彩均衡化算法.cpp

Author: BBuf

PhotoShop Algorithm/PS色彩均衡化算法.cpp

@@ -0,0 +1,33 @@
+Mat Equalizer(Mat src) {
+	int row = src.rows;
+	int col = src.cols;
+	int Count[256] = { 0 };
+	float p[256] = { 0 };
+	float fSum[256] = { 0 };
+	int level[256] = { 0 };
+	int Total = row * col * 3;
+	for (int i = 0; i < row; i++) {
+		for (int j = 0; j < col; j++) {
+			for (int k = 0; k < 3; k++) {
+				Count[src.at<Vec3b>(i, j)[k]]++;
+			}
+		}
+	}
+	for (int i = 0; i < 256; i++) {
+		p[i] = 1.0 * Count[i] / (1.0 * Total);
+		if (i == 0)
+			fSum[0] = p[0];
+		else
+			fSum[i] = fSum[i - 1] + p[i];
+		level[i] = saturate_cast<uchar>(255 * fSum[i] + 0.5);
+	}
+	Mat dst(row, col, CV_8UC3);
+	for (int i = 0; i < row; i++) {
+		for (int j = 0; j < col; j++) {
+			for (int k = 0; k < 3; k++) {
+				dst.at<Vec3b>(i, j)[k] = level[src.at<Vec3b>(i, j)[k]];
+			}
+		}
+	}
+	return dst;
+}

PhotoShop Algorithm/README.md

@@ -26,4 +26,5 @@
 - PSMinMax运算算法.cpp 实现了MinMax运算，对两张输入大小相同的图片进行逐像素运算合成目标图像。
 - PS算术乘法和除法.cpp 实现了对输入的两张图像进行乘法或除法的运算合成新图像。
 - PS算术加法和减法.cpp 实现了对输入的两张图像进行加法或减法的运算合成新图像。
+- PS色彩均衡化算法.cpp 实现了PS色彩均衡化算法，原理请看：https://blog.csdn.net/just_sort/article/details/94430129