OpenCV手动将RGB转换为YCbCr,反之亦然(Visual C ++)

时间:2011-08-07 13:30:23

标签: visual-c++ casting opencv

我想将我的rgb图像转换为ycbcr图像,反之亦然。但是有一个错误,我找不到它。

我的图片看起来不像他们应该的样子。 这是我的代码:

void YCbCrToRGB(IplImage* ScrY, IplImage* ScrCb, IplImage* ScrCr, IplImage* DesR, IplImage* DesG, IplImage* DesB){

    for(int i=0; i < ScrY->height; i++){
        for(int j=0; j < ScrY->width; j++){
            double Y = (double)(ScrY->imageData + ScrY->widthStep*i)[j];
            double Cb = (double)(ScrCb->imageData + ScrCb->widthStep*i)[j];
            double Cr = (double)(ScrCr->imageData + ScrCr->widthStep*i)[j];

            Cb = Cb -128;
            Cr = Cr -128;
            int r;
            int g;
            int b;

            (DesR->imageData + DesR->widthStep*i)[j] = (int)(1 * Y + 0 * Cb + 1.4 * Cr);
            (DesG->imageData + DesG->widthStep*i)[j] = (int)(1 * Y - 0.343 * Cb - 0.711 *Cr);
            (DesB->imageData + DesB->widthStep*i)[j] = (int)(1* Y + 1.765 * Cb + 0* Cr);
    }}

void RGBtoYCbCr(IplImage* ScrR, IplImage* ScrG, IplImage* ScrB, IplImage* DesY, IplImage* DesCb, IplImage* DesCr){
    for(int i=0; i < ScrR->height; i++){
        for(int j=0; j < ScrR->width; j++){
            double R = (double)(ScrR->imageData + ScrR->widthStep*i)[j];
            double G = (double)(ScrG->imageData + ScrG->widthStep*i)[j];
            double B = (double)(ScrB->imageData + ScrB->widthStep*i)[j];

            (DesY->imageData + DesY->widthStep*i)[j]   =       0.299    *R + 0.587  *G  + 0.114  *B;
            (DesCb->imageData + DesCb->widthStep*i)[j] =      -0.1687   *R - 0.3313 *G  + 0.5    *B + 128;
            (DesCr->imageData + DesCr->widthStep*i)[j] =       0.5      *R - 0.4187 *G  - 0.0813 *B + 128;          
        }
    }}

int _tmain(int argc, _TCHAR* argv[])
{
    try {
        IplImage* img = cvLoadImage( "C:\\sad-cat.jpg",1 ); 
        cvNamedWindow( "Example1", CV_WINDOW_AUTOSIZE ); 
        cvShowImage( "Example1", img );

        IplImage *r = cvCreateImage(cvGetSize(img), img->depth, 1);
        IplImage *g = cvCreateImage(cvGetSize(img), img->depth, 1);
        IplImage *b = cvCreateImage(cvGetSize(img), img->depth, 1);

        cvSplit(img, b, g, r, NULL);

        IplImage *y  = cvCreateImage(cvGetSize(img), img->depth, 1);
        IplImage *cb = cvCreateImage(cvGetSize(img), img->depth, 1);
        IplImage *cr = cvCreateImage(cvGetSize(img), img->depth, 1);


        RGBtoYCbCr(r,g,b,y,cb,cr);

        IplImage *ycbcr = cvCreateImage(cvGetSize(img), img->depth, 3);
        cvMerge(y,cb,cr,NULL,ycbcr);

        cvNamedWindow( "YCbCr from RGB", CV_WINDOW_AUTOSIZE ); 
        cvShowImage( "YCbCr from RGB", ycbcr );



        YCbCrToRGB(y,cb,cr,r,g,b);

        IplImage *RGBfromYCbCr = cvCreateImage(cvGetSize(img), img->depth, 3);
        cvMerge(r,g,b,NULL,RGBfromYCbCr);

        cvNamedWindow( "RGB from YCbCr", CV_WINDOW_AUTOSIZE ); 
        cvShowImage( "RGB from YCbCr", RGBfromYCbCr );

        cvWaitKey(0);
        return 0;
    }
    catch( exception& e){
        std::cout<<("An error occurred.") << std::endl;
        std::cout << e.what() <<std::endl; // Print the error message. 
         cvWaitKey(0);
    }  
}

以下是带语法高亮显示的链接:http://paste2.org/p/1569904

我认为这是一个投射错误或类似的事情。 非常感谢你

PS:查看公式和示例图片的评论(我不允许发布两个以上的链接或图片)

2 个答案:

答案 0 :(得分:3)

是的,这是一个投射问题。

        double Y = (double)(ScrY->imageData + ScrY->widthStep*i)[j];
        double Cb = (double)(ScrCb->imageData + ScrCb->widthStep*i)[j];
        double Cr = (double)(ScrCr->imageData + ScrCr->widthStep*i)[j];

应该是

        double Y = (uchar)(ScrY->imageData + ScrY->widthStep*i)[j];
        double Cb = (uchar)(ScrCb->imageData + ScrCb->widthStep*i)[j];
        double Cr = (uchar)(ScrCr->imageData + ScrCr->widthStep*i)[j];

RGBtoYCbCr的前三行中进行相同的更改。

此外,您有舍入错误。您应该在分配之前对结果进行舍入:

        (DesY->imageData + DesY->widthStep*i)[j]   = round(     0.299    *R + 0.587  *G  + 0.114  *B);
        (DesCb->imageData + DesCb->widthStep*i)[j] = round(    -0.1687   *R - 0.3313 *G  + 0.5    *B) + 128;
        (DesCr->imageData + DesCr->widthStep*i)[j] = round(     0.5      *R - 0.4187 *G  - 0.0813 *B) + 128;          

最后,cvMerge(r,g,b,NULL,RGBfromYCbCr);应为cvMerge(b,g,r,NULL,RGBfromYCbCr);,因为cvShowImage假设为BGR,而不是RGB。

答案 1 :(得分:0)

好吧,我的回答可能很愚蠢,但你应该知道cvShowImage()会将它读取的内容解释为BGR图像。如果您转换为YCbCr并尝试查看YCbCr的外观,请确保它看起来很奇怪:

  • Y解释为B频道
  • Cb被解释为G频道
  • Cr解释为R通道

我不知道你对你的代码运行了哪个测试,最好的最终测试是从你的RGBtoYCbCr()的YCbCrToRGB()中读取图像

  
      
  • 您的图片=&gt; RGBToYCbCr()=&gt; YCbCrToRGB()=&gt; cvShowImage()
  •   

第二条建议,检查你是否写过BGR图像,如果不是你需要从RGB转换为BGR ......

告诉我你获得了什么,以及它是否有效,

于连