对于彩色的图片来说,直方图均衡化一般不能直接对R、G、B三个分量分别进行上述的操作,而要将RGB转换成HSV来对V分量进行直方图均衡化的操作。
bool EqualizeHist(Mat &src,Mat &dst)
{
int cn = src.channels();
int w = src.cols;
int h = src.rows;
Mat hsv;
if(cn !=3)
return false;
cvtColor(src,hsv,CV_BGR2HSV);
const int hist_sz = 256;
int hist[hist_sz];
memset(hist, 0, sizeof(hist));
uchar *ptr = hsv.data;
for (int y=0;y<h;y++)
{
for (int x=0;x<w;x++)
{
int idx = y*w*cn + x*cn + 2;
hist[ptr[idx]]++;
}
}
float scale = 255.f/(w*h);
int sum = 0;
uchar lut[hist_sz+1];
for( int i = 0; i < hist_sz; i++ )
{
sum += hist[i];
int val = cvRound(su