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）

最后

以上就是炙热羊为你收集整理的FFmpeg的AVFrame数据转换到OpenCV Mat中的全部内容，希望文章能够帮你解决FFmpeg的AVFrame数据转换到OpenCV Mat中所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。