ffmpeg中avframe的YUV格式数据到OpenCV中Mat的BGR格式转换

ffmpeg实现音视频编解码是非常常用的工具,视频解码出来的raw数据是yuv格式,用来进行后续的图像处理一般是RGB格式的。所以需要从yuv到rgb或者bgr的转换,ffmpeg提供了相应的转换API函数:

下面代码中dec_ctx是解码器上下文,AV_PIX_FMT_BGR24是要转换成的图像数据格式,通过avpicture_get_size()函数获取图像的数据占用空间大小,并使用av_malloc()分配一个outBuff。将outbuff挂到video_frameBGR结构体上,并设置好格式转换上下文sws_getContext()。当然也要用OpenCV声明一个Mat 来保存最后的BGR图像。

    struct SwsContext *pSwsCtx;
    AVFrame *video_frameBGR=NULL;
    video_frameBGR = av_frame_alloc();
    uint8_t *outBuff = NULL;
    int frameSize;
    {
        frameSize = avpicture_get_size(AV_PIX_FMT_BGR24, dec_ctx->width, dec_ctx->height);
        outBuff = (uint8_t*)av_malloc(frameSize);
        avpicture_fill((AVPicture *)video_frameBGR, outBuff, AV_PIX_FMT_BGR24, dec_ctx->width, dec_ctx->height);

        //ÉèÖÃͼÏñת»»ÉÏÏÂÎÄ  
        pSwsCtx = sws_getContext(dec_ctx->width, dec_ctx->height, dec_ctx->pix_fmt,
            dec_ctx->width, dec_ctx->height, AV_PIX_FMT_BGR24,
            SWS_BICUBIC, NULL, NULL, NULL);
    }
    cv::Mat img = cv::Mat::zeros(dec_ctx->height, dec_ctx->width, CV_8UC3);

使用ffmpeg的sws_scale()接口函数实现YUV格式的video_frame到BGR格式的video_frameBGR的转换,数据保存在缓冲outBuff中,从outBuff中拷贝到Mat中就得到一副BGR图像供OpenCV使用。

sws_scale(pSwsCtx, video_frame->data,
                        video_frame->linesize, 0, dec_ctx->height,
                        video_frameBGR->data, video_frameBGR->linesize);

                    memcpy(img.data, outBuff, frameSize);

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏红色石头的机器学习之路

图像凸性检测函数convexityDefects在Python2.7下使用opencv3.0的问题

最近在学习Python下的OpenCV,在图像的凸性检测中,发现opencv3.0下的convexityDefects函数对图像的凸性缺陷处理有错误。不知道是o...

28000
来自专栏生信技能树

使用ESTIMATE来对转录组表达数据根据stromal和immune细胞比例估算肿瘤纯度

ESTIMATE (Estimation of STromal and Immune cells in MAlignant Tumor tissues usin...

85120
来自专栏CVer

利用OpenCV和深度学习实现人脸检测

今天偷点儿懒,就没有及时整理最新的paper,还请各位看官谅解。这里整理了一份前段时间做的小demo,实现献丑了 本文基于OpenCV3.3.1或以上版本(如O...

4.4K150
来自专栏HansBug's Lab

算法模板——单个值欧拉函数

输入N,输出phi(N) 这样的单个值欧拉函数程序一般见于部分数论题,以及有时候求逆元且取模的数不是质数的情况(逆元:A/B=A*Bphi(p)-1 (mod ...

36250
来自专栏一棹烟波

OpenGL进行简单的通用计算实例

博主作为OpenGL新手,最近要用OpenGL进行并行的数据计算,突然发现这样的资料还是很少的,大部分资料和参考书都是讲用OpenGL进行渲染的。好不容易找到一...

32970
来自专栏Java与Android技术栈

利用tess-two和cv4j实现简单的ocr功能、

Tesseract是Ray Smith于1985到1995年间在惠普布里斯托实验室开发的一个OCR引擎,曾经在1995 UNLV精确度测试中名列前茅。但1996...

26510
来自专栏小鹏的专栏

kinect v2.0原理介绍之十一:录制视频

不小心误删了,再加上。 ~~有兴趣的小伙伴,加kinect算法交流群:462964980。 录制视频是结合openCV来做的。  代码如下: // 19...

30190
来自专栏技术随笔

LIDC-IDRI肺结节公开数据集Dicom和XML标注详解数据来源解析结果数据分析

1K80
来自专栏小樱的经验随笔

HDU 1874 畅通工程续【Floyd算法实现】

畅通工程续 Time Limit: 3000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Jav...

302100
来自专栏CVPy

OpenCV检测篇(二):笑脸检测

上篇分享了如何做猫脸检测,本文与之具有知识上的连贯性,所以建议没读过前一篇的先去阅读一下前一篇。这篇主要给大家介绍下如何使用OpenCV进行笑脸检测。

1.6K10

扫码关注云+社区

领取腾讯云代金券