C++自定义直方图均衡化

Mat HistBalance(Mat img) {
	Mat img_dect = img;
	//imshow("原图", img_dect);
	Mat img_balance = img;
	img_dect.at<Vec3b>(0, 0)[0];
	//获得图像大小
	int width = img_dect.cols;
	int hight = img_dect.rows;
	int size_n = width * hight;
	//统计直方图数据分别累加每个颜色值的数量
	float B[256] = { 0.0 };
	float G[256] = { 0.0 };
	float R[256] = { 0.0 };
	for (int i = 0; i < width; i++) {
		for (int j = 0; j < hight; j++)
		{
			int b = img_dect.at<Vec3b>(j, i)[0];
			int g = img_dect.at<Vec3b>(j, i)[1];
			int r = img_dect.at<Vec3b>(j, i)[2];

			B[b] += 1;
			G[g] += 1;
			R[r] += 1;
		}
	}
	//直方图均衡化
	float B_blance[255] = { 0 };
	float G_blance[255] = { 0 };
	float R_blance[255] = { 0 };
	for(int i = 0;i<256;i++)
		for (int j = 0; j <= i; j++) {
			B_blance[i] =(B[j] / size_n) + B_blance[i];
			G_blance[i] = (G[j] / size_n) + G_blance[i];
			R_blance[i] = (R[j] / size_n) + R_blance[i];
		}
	for (int i = 0; i < 256; i++)
	{
		B_blance[i] = 255 * B_blance[i];
		G_blance[i] = 255 * G_blance[i];
		R_blance[i] = 255 * R_blance[i];
	}

	//将均衡化后的直方图赋给Image对象
	for (int i = 0; i < width; i++) {
		for (int j = 0; j < hight; j++)
		{
			img_balance.at<Vec3b>(j,i)[0] = (unsigned)B_blance[img_dect.at<Vec3b>(j, i)[0]];
			img_balance.at<Vec3b>(j, i)[1] = (unsigned)G_blance[img_dect.at<Vec3b>(j, i)[1]];
			img_balance.at<Vec3b>(j, i)[2] = (unsigned)R_blance[img_dect.at<Vec3b>(j, i)[2]];
		}
	}
	return img_balance;
}