OpenCV-白平衡(灰度世界算法)

Posted 2022-05-31 翟天保Steven

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了OpenCV-白平衡(灰度世界算法)相关的知识，希望对你有一定的参考价值。

作者：翟天保Steven
版权声明：著作权归作者所有，商业转载请联系作者获得授权，非商业转载请注明出处

实现原理

白平衡的意义在于，对在特定光源下拍摄时出现的偏色现象，通过加强对应的补色来进行补偿，使白色物体能还原为白色。

灰度世界算法是白平衡各种算法中最基本的一种。它假设图像世界具备丰富色彩，红蓝绿三通道的灰度值在平均后趋近一致，该值作为“灰色”；若各通道均值偏离“灰色”，则将其进行补偿，使其回归“灰色”，进而实现白平衡的效果。

通俗的讲，若图像中绿色较强，蓝色和红色较弱，则用了灰度世界算法后，绿色会适当减弱，蓝色和红色会适当加强，这样就使原本偏色严重的情况得到了缓解。

灰度世界算法的实现流程如下：

1.计算图像RGB三通道各自的灰度平均值Raver、Gaver、Baver。

2.计算“灰色”：Gray=(Raver+Gaver+Baver)/3。

3.计算三通道的补偿系数，即灰色值除以单通道平均值。

功能函数代码

// 白平衡-灰度世界
cv::Mat WhiteBalcane_Gray(cv::Mat src)

	cv::Mat result = src.clone();
	if (src.channels() != 3)
	
		cout << "The number of image channels is not 3." << endl;
		return result;
	

	// 通道分离
	vector<cv::Mat> Channel;
	cv::split(src, Channel);

	// 计算通道灰度值均值
	double Bm = cv::mean(Channel[0])[0];
	double Gm = cv::mean(Channel[1])[0];
	double Rm = cv::mean(Channel[2])[0];
	double Km = (Bm + Gm + Rm) / 3;

	// 通道灰度值调整
	Channel[0] *= Km / Bm;
	Channel[1] *= Km / Gm;
	Channel[2] *= Km / Rm;

	// 合并通道
	cv::merge(Channel, result);

	return result;

C++测试代码

#include <iostream>
#include <opencv.hpp>

using namespace std;

// 白平衡-灰度世界
cv::Mat WhiteBalcane_Gray(cv::Mat src)

	cv::Mat result = src.clone();
	if (src.channels() != 3)
	
		cout << "The number of image channels is not 3." << endl;
		return result;
	

	// 通道分离
	vector<cv::Mat> Channel;
	cv::split(src, Channel);

	// 计算通道灰度值均值
	double Bm = cv::mean(Channel[0])[0];
	double Gm = cv::mean(Channel[1])[0];
	double Rm = cv::mean(Channel[2])[0];
	double Km = (Bm + Gm + Rm) / 3;

	// 通道灰度值调整
	Channel[0] *= Km / Bm;
	Channel[1] *= Km / Gm;
	Channel[2] *= Km / Rm;

	// 合并通道
	cv::merge(Channel, result);

	return result;


int main()

	// 载入原图
	cv::Mat src = cv::imread("test.jpg");

	// 白平衡-灰度世界
	cv::Mat result = WhiteBalcane_Gray(src);

	// 显示
	cv::imshow("src", src);
	cv::imshow("result", result);
	cv::waitKey(0);

	return 0;