C++实现直方图均衡，附源代码

接口设计

//直方图均衡化函数定义
	static BOOL histeq(CImageDataset &imgIn, CImageDataset &imgOut); 

运行结果

源代码

//计算直方图函数
void CImageProcessingEx::getHist(const CImageDataset& img,int band,std::vector<int>& hist)
{
	int value;
	int N_row = img.m_ysize;
	int N_col = img.m_xsize;
	const double* imgData = img.m_data;
	for(int row=0; row<N_row; ++row)  //对图像逐行逐列进行扫描，计算完成直方图
	{ 
		for(int col=0; col<N_col; ++col) 
		{ 
			int index = band * N_row * N_col + row * N_col + col;
			value=(UINT8)imgData[index];
			hist[value]++; 
		} 
	} 
}


//直方图均衡
BOOL CImageProcessingEx::histeq(CImageDataset &imgIn, CImageDataset &imgOut) 
{ 
	const int LEVEL = 256; //灰度值级数
	int k, row, col; //用于循环
	//double hist[LEVEL], accuHist[LEVEL]; //直方图和累积直方图
	std::vector<int> hist(LEVEL);
	std::vector<double> accuHist(LEVEL);

	/* 确保输入图像非空 */ 
	if(imgIn.empty()) 
	{ 
		return FALSE; 
	} 

	//记录输入图像的尺寸
	init(imgIn);

	/*创建输出图像*/ 
	if(FALSE == imgOut.create(n_col,n_row,n_band))
	{ 
		return FALSE; 
	} 

	//输入和输出图像的数据区指针
	const double *input_data=imgIn.m_data;
	double *output_data = imgOut.m_data;
	int img_area = n_col * n_row;//面积，即单个波段的像素个数

	for (int band = 0; band < n_band; band++)
	{/*  步骤 1：计算输入图像的直方图  */
		getHist(imgIn, band,hist);

		/*  步骤 2：计算累积直方图  */
		accuHist[0] = hist[0] / img_area;
		for (k = 1; k < LEVEL - 1; k++) //对直方图索引进行扫描，计算完成累积直方图
		{
			accuHist[k] = accuHist[k - 1] + double(hist[k]) / img_area;
		}
		accuHist[255] = 1;

		/*  步骤 3：根据用累积直方图建立灰度变换函数，并逐行逐列进行像素的灰度变换 */
		int index, gray_value;
		for (row = 0; row < n_row; ++row)
		{
			for (col = 0; col < n_col; ++col)
			{
				index = coo2ind(band,row,col);
				gray_value = (int)input_data[index];

				output_data[index] = (LEVEL - 1) * accuHist[gray_value];//灰度变换
			}
		}
	}
	return TRUE;
}