yuv转opencv中的IplImage

一个小的程序，在网上找了很久没有发现

自己搞了一个大家看看

第一个是很笨的办法：

yuv三个分量分别写在3个矩阵下，然后合并之后转换为rgb分量的图片格式就可以了；

代码如下：

[html] view plaincopyprint?

1. IplImage *image,*rgbimg,*yimg,*uimg,*vimg,*uuimg,*vvimg;
3. rgbimg = cvCreateImage(cvSize(nWidth, nHeight),IPL_DEPTH_8U,3);
4. image = cvCreateImage(cvSize(nWidth, nHeight),IPL_DEPTH_8U,3);
6. yimg = cvCreateImageHeader(cvSize(nWidth, nHeight),IPL_DEPTH_8U,1);
7. uimg = cvCreateImageHeader(cvSize(nWidth/2, nHeight/2),IPL_DEPTH_8U,1);
8. vimg = cvCreateImageHeader(cvSize(nWidth/2, nHeight/2),IPL_DEPTH_8U,1);
10. uuimg = cvCreateImage(cvSize(nWidth, nHeight),IPL_DEPTH_8U,1);
11. vvimg = cvCreateImage(cvSize(nWidth, nHeight),IPL_DEPTH_8U,1);
13. cvSetData(yimg,pBuf, nWidth);
14. cvSetData(uimg,pBuf+nWidth*nHeight, nWidth/2);
15. cvSetData(vimg,pBuf+long(nWidth*nHeight*1.25), nWidth/2);
16. cvResize(uimg,uuimg,CV_INTER_LINEAR);
17. cvResize(vimg,vvimg,CV_INTER_LINEAR);
19. cvMerge(yimg,uuimg,vvimg,NULL,image);
20. cvCvtColor(image,rgbimg,CV_YCrCb2BGR);

IplImage *image,*rgbimg,*yimg,*uimg,*vimg,*uuimg,*vvimg;rgbimg = cvCreateImage(cvSize(nWidth, nHeight),IPL_DEPTH_8U,3);image = cvCreateImage(cvSize(nWidth, nHeight),IPL_DEPTH_8U,3);    yimg = cvCreateImageHeader(cvSize(nWidth, nHeight),IPL_DEPTH_8U,1);uimg = cvCreateImageHeader(cvSize(nWidth/2, nHeight/2),IPL_DEPTH_8U,1);vimg = cvCreateImageHeader(cvSize(nWidth/2, nHeight/2),IPL_DEPTH_8U,1);    uuimg = cvCreateImage(cvSize(nWidth, nHeight),IPL_DEPTH_8U,1);vvimg = cvCreateImage(cvSize(nWidth, nHeight),IPL_DEPTH_8U,1);    cvSetData(yimg,pBuf, nWidth);cvSetData(uimg,pBuf+nWidth*nHeight, nWidth/2);cvSetData(vimg,pBuf+long(nWidth*nHeight*1.25), nWidth/2);cvResize(uimg,uuimg,CV_INTER_LINEAR);cvResize(vimg,vvimg,CV_INTER_LINEAR);    cvMerge(yimg,uuimg,vvimg,NULL,image);cvCvtColor(image,rgbimg,CV_YCrCb2BGR);

还有一个方法就比较负载

首先自己根据原理转换为rgb格式

然后利用cvSetData（）函数写入数据生成IplImage格式的图片

首先定义转换的公式：

[html] view plaincopyprint?

1. #define MR(Y,U,V) (Y + (1.403)*(V-128))
2. #define MG(Y,U,V) (Y - (0.344) * (U-128) - (0.714) * (V-128) )
3. #define MB(Y,U,V) (Y + ((1.773) * (U-128)))

#define MR(Y,U,V) (Y + (1.403)*(V-128))#define MG(Y,U,V) (Y - (0.344) * (U-128) - (0.714) * (V-128) ) #define MB(Y,U,V) (Y + ((1.773) * (U-128)))

yuv转rgb的函数：

[html] view plaincopyprint?

1. void YUV420_C_RGB( char* pYUV, unsigned char* pRGB, int height, int width)
2. {
3. char* pY = pYUV;
4. char* pU = pYUV+height*width;
5. char* pV = pU+(height*width/4);
8. unsigned char* pBGR = NULL;
9. unsigned char R = 0;
10. unsigned char G = 0;
11. unsigned char B = 0;
12. char Y = 0;
13. char U = 0;
14. char V = 0;
15. double tmp = 0;
16. for ( int i = 0; i<height; ++i )
17. {
18. for ( int j = 0; j<width; ++j )
19. {
20. pBGR = pRGB+ i*width*3+j*3;
22. Y = *(pY+i*width+j);
23. U = *pU;
24. V = *pV;
26. //B
27. tmp = MB(Y, U, V);
28. //B = (tmp > 255) ? 255 : (char)tmp;
29. //B = (B<0) ? 0 : B;
30. B = (unsigned char)tmp;
31. //G
32. tmp = MG(Y, U, V);
33. //G = (tmp > 255) ? 255 : (char)tmp;
34. // G = (G<0) ? 0 : G;
35. G = (unsigned char)tmp;
36. //R
37. tmp = MR(Y, U, V);
38. //R = (tmp > 255) ? 255 : (char)tmp;
39. //R = (R<0) ? 0 : R;
40. R = (unsigned char)tmp;
43. *pBGR = R;
44. *(pBGR+1) = G;
45. *(pBGR+2) = B;
48. if ( i%2 == 0 && j%2 == 0)
49. {
50. *pU++;
51. //*pV++;
52. }
53. else
54. {
55. if ( j%2 == 0 )
56. {
57. *pV++ ;
58. }
59. }
60. }
62. }
63. }

void YUV420_C_RGB( char* pYUV, unsigned char* pRGB, int height, int width){    char* pY = pYUV;    char* pU = pYUV+height*width;    char* pV = pU+(height*width/4);    unsigned char* pBGR = NULL;    unsigned char R = 0;    unsigned char G = 0;    unsigned char B = 0;    char Y = 0;    char U = 0;    char V = 0;    double tmp = 0;    for ( int i = 0; i < height; ++i )    {        for ( int j = 0; j < width; ++j )        {            pBGR = pRGB+ i*width*3+j*3;            Y = *(pY+i*width+j);            U = *pU;            V = *pV;            //B            tmp = MB(Y, U, V);            //B = (tmp > 255) ? 255 : (char)tmp;            //B = (B<0) ? 0 : B;B = (unsigned char)tmp;            //G            tmp = MG(Y, U, V);            //G = (tmp > 255) ? 255 : (char)tmp;           // G = (G<0) ? 0 : G;G = (unsigned char)tmp;            //R            tmp = MR(Y, U, V);            //R = (tmp > 255) ? 255 : (char)tmp;            //R = (R<0) ? 0 : R;R = (unsigned char)tmp;            *pBGR     = R;                        *(pBGR+1) = G;                    *(pBGR+2) = B;                    if ( i%2 == 0 && j%2 == 0)            {                *pU++;//*pV++;            }            else            {                if ( j%2 == 0 )                {                    *pV++ ;                }}        }        }}

最后是写入IplImage的代码：

[html] view plaincopyprint?

1. unsigned char* pRGB = NULL;
3. pRGB = (unsigned char*)malloc(nSize*sizeof(unsigned char*)*2);
5. YUV420_C_RGB(pBuf,pRGB,nWidth,nHeight);
7. IplImage *image;
8. image = cvCreateImageHeader(cvSize(nWidth, nHeight),IPL_DEPTH_8U,3);
9. cvSetData(image,pRGB,nWidth*3);

unsigned char* pRGB = NULL;pRGB = (unsigned char*)malloc(nSize*sizeof(unsigned char*)*2);YUV420_C_RGB(pBuf,pRGB,nWidth,nHeight);IplImage *image;image = cvCreateImageHeader(cvSize(nWidth, nHeight),IPL_DEPTH_8U,3);cvSetData(image,pRGB,nWidth*3);

程序都运行过

编译环境为vs2008

opencv2.0版本

另外，附上RGB转YUV420格式的代码：

#define MY(R,G,B) ((0.299)*R + (0.587)*G + (0.114)*B)

#define MU(R,G,B) (-(0.169)*R - (0.332)*G+ (0.501)*B)

#define MV(R,G,B) ( (0.450)*R - (0.419)*G - (0.081)*B )

#define MR(Y,U,V) (Y + (1.403)*V)

#define MG(Y,U,V) (Y - ( (0.3455) * U - ( (0.7169) * V ) ) )

#define MB(Y,U,V) (Y + ((1.7790) * U))

/**************************rgb->yuv420*************************************/

voidRGB_C_YUV420(char* pRGB, char* pYUV, intheight,intwidth)

{

char* pY = pYUV;

char* pU = pYUV+height*width;

char* pV = pU+(height*width/4);

char* pBGR = NULL;

charR = 0;

charG = 0;

charB = 0;

charY = 0;

charU = 0;

charV = 0;

doubletmp = 0;

for(inti = 0; i < height; ++i )

{

for(intj = 0; j < width; ++j )

{

pBGR = pRGB+ i*width*3+j*3;

B = *pBGR;

G = *(pBGR+1);

R = *(pBGR+2);

//Y

Y = MY(R,G,B);

//U

U = MU(R,G,B);

//V

V = MV(R,G,B);

*(pY+i*width+j) = Y;

if( i%2 == 0 && j%2 == 0)

{

*pU = U;

pU++;

}

else

{

if( j%2 == 0 )

{

*pV++ = V;

}

}

}

}

}