我是靠谱客的博主 炙热羊,这篇文章主要介绍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内容请搜索靠谱客的其他文章。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(40)

评论列表共有 0 条评论

立即
投稿
返回
顶部