iOS ffmpeg获取视频关键帧

最新推荐文章于 2023-11-14 14:01:17 发布

番薯大佬

最新推荐文章于 2023-11-14 14:01:17 发布

阅读量1.8k

点赞数

分类专栏： iOS 开发编码收集

iOS 开发编码收集专栏收录该内容

236 篇文章

订阅专栏

本文介绍如何使用FFmpeg SDK从视频中提取关键帧并保存为图片。首先通过av_open_input_file打开视频文件，并找到视频流。接着选择合适的解码器进行解码，将解码后的YUV数据转换为RGB数据，最后将关键帧保存为bmp格式。

av_register_all();   
   
if(av_open_input_file(&pFormatCtx, filename, NULL, 0, NULL)!=0) 
{  
        printf("error!\n"); 
}  
   
if(av_find_stream_info(pFormatCtx)<0)   
{
        printf("error!\n");   
}
   
videoStream=-1;   
   
for(i=0; i<pFormatCtx->nb_streams; i++)   
{
        if(pFormatCtx->streams[i]->codec->codec_type==CODEC_TYPE_VIDEO)   
        {   
            videoStream=i;   
            break;   
        }   
	if(videoStream==-1)   
	{
		printf("error!\n");// Didn't find a video stream  
	} 
   
    // 得到视频流编码上下文的指针   
    pCodecCtx=pFormatCtx->streams[videoStream]->codec;  
}

用两种方式，一是利用ffmpeg提供的可执行文件进行提取，另外就是用ffmpeg的sdk，进行开发。我下面说一下如何使用ffmpeg sdk进行提取（假设把提取的关键帧保存成bmp，源文件名是sample.mpg）：

首先获取文件中的视频流：

然后选择解码器进行解码：

AVCodec *pCodec;

//  寻找视频流的解码器   
pCodec=avcodec_find_decoder(pCodecCtx->codec_id);   
if(pCodec==NULL)  
{ 
    printf("error!\n");// 找不到解码器   
}

// 打开解码器   
if(avcodec_open(pCodecCtx, pCodec)<0)   
{
    printf("error!\n"); // 打不开解码器  
}

现在开始，进入解码和提取关键帧的过程：

pFrame=avcodec_alloc_frame();   
pFrameRGB = avcodec_alloc_frame();   
numBytes=avpicture_get_size(PIX_FMT_BGR24, pCodecCtx->width,pCodecCtx->height);   
buffer=new uint8_t[numBytes];   
avpicture_fill((AVPicture *)pFrameRGB, buffer, PIX_FMT_RGB24,pCodecCtx->width, pCodecCtx->height);   
pSWSCtx = sws_getContext(pCodecCtx->width, pCodecCtx->height, pCodecCtx->pix_fmt, pCodecCtx->width, pCodecCtx->height, PIX_FMT_RGB24, SWS_BICUBIC, NULL, NULL, NULL);   
   
i=0;   
while(av_read_frame(pFormatCtx,&packet)>=0)   
{   
    if(packet.stream_index==videoStream)   
    {   
        avcodec_decode_video(pCodecCtx, pFrame, &frameFinished,packet.data, packet.size);   
        if(frameFinished)   
        {   
            if(pFrame->key_frame==1) // 这就是关键帧   
            {   
                sws_scale(pSWSCtx, pFrame->data, pFrame->linesize,0, pCodecCtx->height, pFrameRGB->data, pFrameRGB->linesize);   
                // 保存到磁盘   
                char pic[200];   
                sprintf(pic,"pic%d.bmp",i);   
                i++;   
                 av_create_bmp(pic,pFrameRGB->data[0],pCodecCtx->width,pCodecCtx->height,24);   
            }   
        }   
    }   
    av_free_packet(&packet);   
}

最后，释放资源和句柄

// 释放 RGB 图象   
av_free(pFrameRGB);

// 释放YUV 帧   
av_free(pFrame);

sws_freeContext(pSWSCtx);

// 关闭解码器（codec）   
avcodec_close(pCodecCtx);

// 关闭视频文件   
av_close_input_file(pFormatCtx);

ffmpeg相关知识

转自：http://fengqing888.blog.163.com/blog/static/330114162011111632548120/