opencv 彩色图像对比度增强

一般地，图像对比度都是在灰度图上进行增强，这在我的一篇文章中已经给出了实现方法（http://blog.csdn.net/abcjennifer/article/details/7401921），最通常的办法就是直方图均衡化，而彩色图像的对比度增强其实道理相同。这里给出我的实现方法：

首先将RGB图像转到YCbCr分量，然后对Y分量上的图像进行直方图均衡化，最后进行图像合成。详见代码~(*^__^*) 

[cpp]  view plain copy

1. /* 
2. *@Function: Color image contrast enhancement 
3. *@Date: 2012-4-5 
4. *@Author: 张睿卿 
5. */  
6.   
7. int ImageStretchByHistogram(IplImage *src1,IplImage *dst1)  
8. /************************************************* 
9. Function:      通过直方图变换进行图像增强，将图像灰度的域值拉伸到0-255 
10. src1:               单通道灰度图像                   
11. dst1:              同样大小的单通道灰度图像  
12. *************************************************/  
13. {  
14.     assert(src1->width==dst1->width);  
15.     double p[256],p1[256],num[256];  
16.   
17.     memset(p,0,sizeof(p));  
18.     memset(p1,0,sizeof(p1));  
19.     memset(num,0,sizeof(num));  
20.     int height=src1->height;  
21.     int width=src1->width;  
22.     long wMulh = height * width;  
23.   
24.     //statistics  
25.     for(int x=0;x<src1->width;x++)  
26.     {  
27.         for(int y=0;y<src1-> height;y++){  
28.             uchar v=((uchar*)(src1->imageData + src1->widthStep*y))[x];  
29.             num[v]++;  
30.         }  
31.     }  
32.     //calculate probability  
33.     for(int i=0;i<256;i++)  
34.     {  
35.         p[i]=num[i]/wMulh;  
36.     }  
37.   
38.     //p1[i]=sum(p[j]);  j<=i;  
39.     for(int i=0;i<256;i++)  
40.     {  
41.         for(int k=0;k<=i;k++)  
42.             p1[i]+=p[k];  
43.     }  
44.   
45.     // histogram transformation  
46.     for(int x=0;x<src1->width;x++)  
47.     {  
48.         for(int y=0;y<src1-> height;y++){  
49.             uchar v=((uchar*)(src1->imageData + src1->widthStep*y))[x];  
50.             ((uchar*)(dst1->imageData + dst1->widthStep*y))[x]= p1[v]*255+0.5;              
51.         }  
52.     }  
53.     return 0;  
54. }  
55.   
56. void CCVMFCView::OnYcbcrY()  
57. {  
58.     IplImage* Y = cvCreateImage(cvGetSize(workImg),IPL_DEPTH_8U,1);  
59.     IplImage* Cb= cvCreateImage(cvGetSize(workImg),IPL_DEPTH_8U,1);  
60.     IplImage* Cr = cvCreateImage(cvGetSize(workImg),IPL_DEPTH_8U,1);  
61.     IplImage* Compile_YCbCr= cvCreateImage(cvGetSize(workImg),IPL_DEPTH_8U,3);  
62.     IplImage* dst1=cvCreateImage(cvGetSize(workImg),IPL_DEPTH_8U,3);  
63.   
64.     int i;  
65.     cvCvtColor(workImg,dst1,CV_BGR2YCrCb);  
66.     cvSplit(dst1,Y,Cb,Cr,0);  
67.   
68.     ImageStretchByHistogram(Y,dst1);  
69.    
70.     for(int x=0;x<workImg->height;x++)  
71.     {  
72.         for(int y=0;y<workImg->width;y++)  
73.         {  
74.             CvMat* cur=cvCreateMat(3,1,CV_32F);  
75.             cvmSet(cur,0,0,((uchar*)(dst1->imageData+x*dst1->widthStep))[y]);  
76.             cvmSet(cur,1,0,((uchar*)(Cb->imageData+x*Cb->widthStep))[y]);  
77.             cvmSet(cur,2,0,((uchar*)(Cr->imageData+x*Cr->widthStep))[y]);  
78.    
79.             for(i=0;i<3;i++)  
80.             {  
81.                 double xx=cvmGet(cur,i,0);  
82.                 ((uchar*)Compile_YCbCr->imageData+x*Compile_YCbCr->widthStep)[y*3+i]=xx;  
83.             }  
84.         }  
85.     }  
86.    
87.     cvCvtColor(Compile_YCbCr,workImg,CV_YCrCb2BGR);  
88.     m_ImageType=3;  
89.     Invalidate();  
90. }