YUV420转RGB颜色转换错误

5

我正在将一个YUV420格式的图像转换为opencv中的RGB图像,但是在转换后得到了一个橙色的图像。我使用了以下代码来进行转换。请问我的代码有什么问题吗?

 int step = origImage->widthStep;
 uchar *data = (uchar *)origImage->imageData; 
 int size = origImage->width * origImage->height;
 IplImage* img1 = cvCreateImage(cvGetSize(origImage), IPL_DEPTH_8U, 3);

for (int i = 0; i<origImage->height; i++)
{
  for (int j=0; j<origImage->width; j++)
  {
    float Y = data[i*step + j];
    float U = data[ (int)(size + (i/2)*(step/2)  + j/2) ];
    float V = data[ (int)(size*1.25 + (i/2)*(step/2) + j/2)];

    float R = Y + (int)(1.772f*V); 
    float G = Y - (int)(0.344f*V + 0.714f*U);
    float B = Y + (int)(1.402f*U);

    if (R < 0){ R = 0; } if (G < 0){ G = 0; } if (B < 0){ B = 0; }
    if (R > 255 ){ R = 255; } if (G > 255) { G = 255; } if (B > 255) { B = 255; }

    cvSet2D(img1, i, j,cvScalar(B,G,R));
  }
}

origImage -> YUV图像,
img1 -> RGB图像。

http://upload.wikimedia.org/wikipedia/en/0/0d/Yuv420.svg

是否有任何OpenCV函数可以将YUV420格式的像素转换为对应的RGB像素?(不是整个图像)

2个回答

12

我通过修改计算 RGB 值的公式得到了答案,这段代码运行良好。

 int step = origImage->widthStep;
 uchar *data = (uchar *)origImage->imageData; 
 int size = origImage->width * origImage->height;
 IplImage* img1 = cvCreateImage(cvGetSize(origImage), IPL_DEPTH_8U, 3);

for (int i = 0; i<origImage->height; i++)
{
  for (int j=0; j<origImage->width; j++)
  {
    float Y = data[i*step + j];
    float U = data[ (int)(size + (i/2)*(step/2)  + j/2) ];
    float V = data[ (int)(size*1.25 + (i/2)*(step/2) + j/2)];

    float R = Y + 1.402 * (V - 128);
    float G = Y - 0.344 * (U - 128) - 0.714 * (V - 128);
    float B = Y + 1.772 * (U - 128);


    if (R < 0){ R = 0; } if (G < 0){ G = 0; } if (B < 0){ B = 0; }
    if (R > 255 ){ R = 255; } if (G > 255) { G = 255; } if (B > 255) { B = 255; }

    cvSet2D(img1, i, j,cvScalar(B,G,R));
  }
}

3

第一个问题是使用已过时的C-API(已经停止更新,请改用C ++)。

第二个问题是编写自己的(缓慢且容易出错的)像素循环。

为什么不使用:

cvtColor(crs,dst, CV_YUV2BGR); // or CV_YUV2BGR_I420

取而代之?


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接