Question

如何强制ffmpeg编码时输出一个关键帧

Answer 1

如何强制ffmpeg编码时输出一个关键帧
AVCodecContext *c //编码器环境句柄
AVFrame* f //需要编码的一帧视频

/*在avcodec.h文件中有这样的定义
#define FF_I_TYPE 1 ///< Intra
#define FF_P_TYPE 2 ///< Predicted
#define FF_B_TYPE 3 ///< Bi-dir predicted
#define FF_S_TYPE 4 ///< S(GMC)-VOP MPEG4
#define FF_SI_TYPE 5 ///< Switching Intra
#define FF_SP_TYPE 6 ///< Switching Predicted
#define FF_BI_TYPE 7
*/

在编码前设置
f->pict_type=FF_I_TYPE;
f->key_frame=1;
注：该帧为I帧时，f->pict_type＝=FF_I_TYPE && f->key_frame==1
然后编码
*outsize = avcodec_encode_video(c, temp, outbuf_size, f);

则编码之后通过如下参数判断是否为关键帧：
key_frame=c->coded_frame->key_frame;
pict_type=c->coded_frame->pict_type;
key_frame==FF_I_TYPE && pict_type==1

ffmpeg如何提取视频的关键帧
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;
然后选择解码器进行解码：
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 SDK就支持，以下代码是ffmpeg官方小组提供的

int main()
{
SwsContext *pSWSCtx;
AVFormatContext *pFormatCtx;
const char *filename="sample.mpg";
int i,videoStream,y_size;
AVCodecContext *pCodecCtx;
AVFrame *pFrame;
AVFrame *pFrameRGB;
int numBytes,frameFinished;
uint8_t *buffer;
static AVPacket packet;

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");

dump_format(pFormatCtx, 0, filename, false);

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;
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);
i++;
}
}
}
av_free_packet(&packet);
}

av_free(pFrameRGB);

av_free(pFrame);

sws_freeContext(pSWSCtx);

avcodec_close(pCodecCtx);

av_close_input_file(pFormatCtx);

return 0;

}

live555+ffmpeg如何提取关键帧（I帧，P帧，B帧）

live555+ffmpeg如何提取关键帧（I帧，P帧，B帧）开发流媒体播放器的时候，特别是在windows mobile,symbian(S60)平台开发时，很可能遇到需要自己开发播放器的情况。S60平台提供了CVideoPlayUtility接口可以实现流媒体播放器，但由于非开源，所以相对于自己开发播放器，很多操作受到限制。live555主要用于网络流接收，ffmpeg则是对接收到的数据进行编码/解码。I帧，P帧，B帧是视频流中三种分类，其中I帧也就是关键帧是基础帧，P帧一般根据I帧确定，而B帧需要前面两着的信息。举例说：the Input sequence for video encoder1 2 3 4 5 6 7I B B P B B I Let's take 1,2,3.. as PTS for simplification the out sequence for video encoder ( this equals the decoder sequence)1 4 2 3 7 5 6I P B B I B B 播放器LIVE555收到的序列顺序就应该是： 1 4 2 3 7 5 6 经过解码器解码，顺序又回到1 2 3 4 5 6 7这种正常顺序。 所以我们可以根据avcodec_decode_video来判断帧别。avcodec_decode_video之后的顺序是一定的。严格按照1 2 3 4。。。这样的顺序来。判断I帧,P,B帧方法:(1):假如解码成功，则不是I帧就是P帧(根据AVFrame->keyframe判断是否是I帧)。假如不是I帧，也不是P帧，则只能是B帧(通过pts判断)。(2):采用AVFrame->pict_type综合pts的办法:if(FF_I_TYPE==picture->pict_type) { Printlog("<II>"); } else if(FF_P_TYPE==picture->pict_type) { Printlog("<PP>"); } else if(FF_B_TYPE==picture->pict_type) { Printlog("<BB>"); } else if(FF_S_TYPE==picture->pict_type) { Printlog("<SS>"); } else { Printlog("<OtherType>"); }正常情况下是不会打印出B帧的,因为解码成功的肯定是I帧或者是P帧.