FFmpeg的AVFrame数据转换到OpenCV Mat中

使用FFmpeg的视频解码功能,可以接受rtsp 的H264码流,得到YUV数据;

解码后的数据放在AVFrame 结构中,

可参考雷神的博客https://blog.csdn.net/leixiaohua1020/article/details/14214577

现将YUV数据转到Mat中

                 AVFrame *frame;

                int width = frame->width, height = frame->height;
                cv::Mat tmp_img = cv::Mat::zeros( height*3/2, width, CV_8UC1 );    
                memcpy( tmp_img.data, frame->data[0], width*height );
                memcpy( tmp_img.data + width*height, frame->data[1], width*height/4 );
                memcpy( tmp_img.data + width*height*5/4, frame->data[2], width*height/4 );

                cv::imshow( "yuv_show", tmp_img );
                cv::Mat bgr;
                cv::cvtColor( tmp_img, bgr, CV_YUV2BGR_I420 );
                cv::imshow( "bgr_show", bgr );
                cv::waitKey(1);

需要讲解的是:Mat的rows是frame->heigth的 3/2倍,cols等于frame->width,单通道;

如果要转换成BGR显示,用CV_YUV2BGR_I420;  因为rtsp流就是IYUV(I420)格式;

通过查看效果,YUV转BGR后与原视频一样。

cv::cvtColor( tmp_img, bgr, CV_YUV2BGR_I420 );

CV_BGR2YUV_I420:BGR转换成IYUV  即I420;

CV_YUV2BGR_I420:从I420  转换成BGR(24)

猜你喜欢

转载自blog.csdn.net/cyf15238622067/article/details/82769735