转换为YCbCr空间的代码

转换为YCbCr空间的代码

2008-11-02 14:03

#include

#include

int main(int argc, char *argv[])

{

IplImage* img = 0; // 原图

IplImage* dst_img;//ycbcr图

IplImage* dst;//肤色图

//IplImage* dst2;//2值

//IplImage* dst_img;

//IplImage* dst2;

img=cvLoadImage("2.jpg",-1);//加载图像

dst=cvCreateImage(cvSize(img->width,img->height),IPL_DEPTH_8U,1);

//dst2=cvCreateImage(cvSize(img->width,img->height),IPL_DEPTH_8U,1); //cvZero(dst2);

dst_img=cvCreateImage(cvSize(img->width,img->height),IPL_DEPTH_8U,3); int height,width,step,channels;

uchar *data,*data1;

height = dst_img->height;

width = dst_img->width;

step = dst_img->widthStep;

channels = dst_img->nChannels;

data = (uchar *)dst_img->imageData;

data1 = (uchar *)img->imageData;

cvCvtColor(img, dst_img, CV_RGB2YCrCb);// rgb->ycrcb

double c[4]={0.0077 ,-0.0041,-0.0041 ,0.0047};//(肤色)协方差矩阵求逆后{0.00626418 ,-0.000254013,-0.000254013 ,0.00334967}{0.0077 ,-0.0041,-0.0041 ,0.0047}

CvMat* max=cvCreateMat(height,width,CV_64FC1);//存储肤色概率密度矩阵（数组）

CvMat* mat=cvCreateMat(2,2,CV_64FC1);//存储协方差矩阵（逆）

CvMat* a=cvCreateMat(1,2,CV_64FC1);//{cb,cr}

CvMat* a1=cvCreateMat(2,1,CV_64FC1);//{cb,cr}转至

CvMat* res=cvCreateMat(1,1,CV_64FC1);

cvInitMatHeader(mat,2,2,CV_64FC1,c);

double res1[1]={0};

cvInitMatHeader(res,1,1,CV_64FC1,res1,CV_AUTOSTEP);

cvNamedWindow("rgb", CV_WINDOW_AUTOSIZE);

cvNamedWindow("肤色", CV_WINDOW_AUTOSIZE);

cvNamedWindow("ycbcr", CV_WINDOW_AUTOSIZE);

int i,j;

double b,r,cb,cr,u;

//按照上面的公式进行肤色似然度计算

for(i=0;i

for(j=0;j

{

b=data[i*step+j*channels+1];//cb 分量

r=data[i*step+j*channels+2];//cr 分量

cb=b-103.0056;////117.4361

cr=r-140.1309;////156.5599

double p1[2]={cb,cr};

cvInitMatHeader(a,1,2,CV_64FC1,p1,CV_AUTOSTEP); cvInitMatHeader(mat,2,2,CV_64FC1,c,CV_AUTOSTEP); cvMatMulAdd(a,mat,0,a);

double p2[2]={cb,cr};/////////////////////////////////////// cvInitMatHeader(a1,2,1,CV_64FC1,p2);

cvMatMulAdd(a,a1,0,res);

u=CV_MAT_ELEM(*res,double,0,0);

u=exp(-0.5*u);

cvmSet(max,i,j,u);

}

double max1=0; //初始化最大值

double mm=0;

cvMinMaxLoc(max,NULL,&max1,NULL,NULL);

//肤色似然度

for(i=0;i

{

mm=CV_MAT_ELEM(*max,double,i,j)/max1;

cvmSet(max,i,j,mm);

((dst->imageData + dst->widthStep*i))[j] = 255*mm;

}

cvSmooth( dst, dst,CV_MEDIAN,3, 0, 0 );//中值滤波cvShowImage("rgb", img );

cvShowImage("ycbcr", dst_img );

cvShowImage("肤色", dst );

cvWaitKey(0);

cvReleaseImage(&img );

return 0; }